작성한 질문수
[개정판] 딥러닝 컴퓨터 비전 완벽 가이드
작성
·
151
0
혹시 batchsize는 어떻게 설정이 가능할까요?? 어느 config를 바꿔야하는지 잘모르겠네요.
답변 1
안녕하십니까,
samples_per_gpu 설정을 변경하시면 됩니다. samples_per_gpu는 GPU 카드 한장당 batch_size를 의미합니다.
cfg.data.samples_per_gpu = 4 이면 gpu 당 batch size를 4로 설정합니다.
감사합니다.
