배치사이즈 썸네일형 리스트형 [CV 문서 분류 대회] 배치 크기(batch size)는 어떻게 결정해야 하나 배치 사이즈는 실험하는 사람이 결정할 수 있는 파라미터 중 하나로, 최적값은 아래 3가지 요소를 보고 결정해야 한다. 1) GPU 메모리 크기: 가장 중요한 제한 요소2) 모델 크기: 더 큰 모델은 더 많은 메모리 필요3) 이미지 크기: 큰 이미지는 더 많은 메모리 사용GPU 메모리 크기를 넘으면 RuntimeError: CUDA out of memory 이런 에러가 뜬다.일반적으로 배치 크기는 16, 32, 64, 128, 256 이런 식으로 많이 쓰는데, 실험적으로 접근해보면 좋다.# 배치 크기별 실험 예시batch_sizes = [16, 32, 64, 128]results = {}for bs in batch_sizes: BATCH_SIZE = bs trn_loader = DataLoade.. 더보기 이전 1 다음