ultralytics / yolov5

YOLOv5 🚀 in PyTorch > ONNX > CoreML > TFLite
https://docs.ultralytics.com
GNU Affero General Public License v3.0
50.19k stars 16.2k forks source link

Reproducibility in multi-process data loading #8601

Closed UnglvKitDe closed 2 years ago

UnglvKitDe commented 2 years ago

Search before asking

Description

In the official documentation it is recommended to set the rseed for the worker to ensure reproducibility even for multi-process dataloaders.

Use case

No response

Additional

No response

Are you willing to submit a PR?

glenn-jocher commented 2 years ago

@UnglvKitDe I'm seeing fully reproducible results now with master and torch>=1.12.0. Is this not the case for you?

Screen Shot 2022-07-19 at 6 21 53 PM