/smddp-yolov5-example

Distributed data parallel YOLOv5 training with PyTorch and SageMaker distributed.

Primary LanguageJupyter NotebookBSD 2-Clause "Simplified" LicenseBSD-2-Clause

smddp-yolov5-example

Distributed data parallel YOLOv5 training with PyTorch and SageMaker distributed. Select us-west-2 as your region. Run this code in SageMaker Notebook instance, and open yolov5.ipynb with conda_pytorch_p39.