Small edits (#901)

This commit is contained in:
pollfly
2024-08-12 16:04:50 +03:00
committed by GitHub
parent 3021efd45c
commit 8b94fa6c1a
17 changed files with 67 additions and 58 deletions

View File

@@ -137,7 +137,7 @@ task = Task.init(
# Load a model
model_variant = "yolov8n"
# Log "model_variant" parameter to task
task.set_parameter("model_variant", model_variant)
task.set_parameter(name="model_variant", value=model_variant)
# Load the YOLOv8 model
model = YOLO(f'{model_variant}.pt')