• 카테고리

    질문 & 답변
  • 세부 분야

    컴퓨터 비전

  • 해결 여부

    미해결

mmdetect시

22.09.13 11:09 작성 조회수 101

0

혹시 batchsize는 어떻게 설정이 가능할까요?? 어느 config를 바꿔야하는지 잘모르겠네요.

답변 1

답변을 작성해보세요.

0

안녕하십니까,

samples_per_gpu 설정을 변경하시면 됩니다. samples_per_gpu는 GPU 카드 한장당 batch_size를 의미합니다.

cfg.data.samples_per_gpu = 4 이면 gpu 당 batch size를 4로 설정합니다.

감사합니다.