diff --git a/supervisely/export_weights/config.json b/supervisely/export_weights/config.json index 9874d2a6f460..5116f6e09eb4 100644 --- a/supervisely/export_weights/config.json +++ b/supervisely/export_weights/config.json @@ -10,7 +10,7 @@ "nn tools" ], "description": "to TorchScript and ONNX formats", - "docker_image": "supervisely/yolov5:1.0.0", + "docker_image": "supervisely/yolov5:1.0.2", "instance_version": "6.8.54", "main_script": "supervisely/export_weights/src/sly_export_weights.py", "task_location": "workspace_tasks",