728x90
반응형
https://github.com/ultralytics/yolov5/issues/475
Multi-GPU Training 🌟 · Issue #475 · ultralytics/yolov5
📚 This guide explains how to properly use multiple GPUs to train a dataset with YOLOv5 🚀 on single or multiple machine(s). UPDATED 18 August 2021. Before You Start Clone repo and install requiremen...
github.com
1. DataParallel
시간 확인 : 0.293 hours
2. DistributedDataParallel
시간 확인 : 0.269 hours
DistributedDataParallel 를 추천 한다고 되여 있는데 시간도 빠르다.
DataParallel와 DistributedDataParallel 의 비교 :
https://algopoolja.tistory.com/56
torch.distributed
torch로 병렬화를 하기 위해서 torch에서 제안하는 몇가지 선택할 수 있는 선택지가 있다. 1. torch.nn.DataParallel DataParallel은 하나의 본체에서 multi-GPU 병렬화를 코딩을 많이 하지 않고 할 수 있는 선택.
algopoolja.tistory.com
반응형
'Deep learning > 소스' 카테고리의 다른 글
| Yolov7 학습하기 (0) | 2022.10.21 |
|---|---|
| yolov5 coco data set training (0) | 2021.09.17 |
| classification mnist-LeNet-5 (0) | 2021.04.03 |
| yolov5 모델pt (0) | 2021.03.26 |
| classification model code (0) | 2021.01.14 |