Skip to content

Conversation

@glenn-jocher
Copy link
Member

@glenn-jocher glenn-jocher commented May 21, 2021

Improved error handling when users combine incompatible --image-weights argument with DDP. Resolves issues #2461 and #3265.

🛠️ PR Summary

Made with ❤️ by Ultralytics Actions

🌟 Summary

Improved validation of training options in YOLOv5 DDP (Distributed Data Parallel) training mode.

📊 Key Changes

  • Added a check to ensure the --image-weights argument is not used with DDP training.

🎯 Purpose & Impact

  • This change ensures the user is informed about incompatible training options, which helps prevent potential errors during the distributed training process.
  • This clarification could lead to smoother user experiences and fewer issues for developers working on training models at scale. 🔄

@glenn-jocher glenn-jocher merged commit 10d56d7 into master May 21, 2021
@glenn-jocher glenn-jocher deleted the glenn-jocher-patch-1 branch May 21, 2021 12:46
@glenn-jocher glenn-jocher self-assigned this May 21, 2021
Lechtr pushed a commit to Lechtr/yolov5 that referenced this pull request Jul 20, 2021
BjarneKuehl pushed a commit to fhkiel-mlaip/yolov5 that referenced this pull request Aug 26, 2022
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

runtimeerror: tensors must be cuda and dense error with weighted image selection for training Multi-GPU Training error

1 participant