딥러닝

[MMdetection] samples_per_gpu

승무_ 2022. 9. 16. 12:03

samples_per_gpu는 gpu 한장당 batch_size입니다

ex) sample_per_gpu = 8 이고 gpu가 2개이면 MMdetection은 batch size를 2*8로 설정합니다.

 

workers_per_gpu는 pytorch의 DataLoader와 연관된 파라미터 입니다. pytorch는 DataLoader를 통해서 CPU의 Data를 GPU로 넘기는데, 이때 병렬 프로세스를 몇개를 띄울 것인가(즉 몇개의 CPU Core 병렬로 사용할 것인가)를 설정하는 파라미터 입니다. DataLoader의 num_workers와 동일 설정입니다. 

'딥러닝' 카테고리의 다른 글

[MMdetection] Grad-CAM  (0) 2022.09.24
[MMdetection] AP for each class  (0) 2022.09.21
convert_to_coco  (0) 2022.09.14
[Colab] unzip  (0) 2022.09.14
[Colab] MMdetection Custom Data 학습  (1) 2022.09.10