딥러닝
[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와 동일 설정입니다.