inflearn logo
강의

강의

N
챌린지

챌린지

멘토링

멘토링

N
클립

클립

로드맵

로드맵

지식공유

[개정판] 딥러닝 컴퓨터 비전 완벽 가이드

mmdetection config 질문드립니다.

1522

lee seungmin

작성한 질문수 8

0

data = dict(
samples_per_gpu=2,
workers_per_gpu=2,
train=dict(
type=dataset_type,
ann_file=data_root + 'annotations/instances_train2017.json',
img_prefix=data_root + 'train2017/',
pipeline=train_pipeline),

 

안녕하세요. 선생님 
config 관련 정보를 보다가 samples_per_gpu 와 workers_per_gpu에 대해서 
알고 싶어 질문드립니다.

GPU 하나를 쓴다고 하였을 때,
samples_per_gpu의 경우는 batch_size와 동일해진다고 해석을 하였는데...
이 해석이 맞는지 헷갈려서 문의드립니다.

그리고 workers_per_gpu 부분은 정확히 무슨 의미인지, 그리고 숫자 증가에 따라 어떤 차이가 발생하고
어떻게 선정을 하는 것인지 궁금합니다.

단순 구글 검색해서 nums_worker 같은 경우, cpu에서 gpu로 데이터를 넘기는데 
cpu 코어를 얼마나 할당해서 gpu로 데이터를 넘길 것인가 부분이라고 나왔는데
이 내용이 config의 workers_per_gpu랑은 다른 내용인 것 같아 질문드립니다.

항상 좋은 강의 감사드립니다.  

mmdetection 머신러닝 배워볼래요? config 딥러닝 samples_per_gpu keras tensorflow python 컴퓨터-비전

답변 1

1

권 철민

안녕하십니까, 

1. 네, samples_per_gpu는 GPU 카드 한장당 batch_size를 의미합니다. 

samples_per_gpu = 8 이고 GPU 카드가 2장이면 MMDetection은 batch size로 2*8=16을 설정합니다. 

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

감사합니다.  

강의 환경설정 질문

0

41

2

Custom Dataset에서의 polygon 정보 관련

0

84

3

cvat.ai 보안 수준이 궁금합니다

0

80

2

캐클 nucleus 챌린지 runpod 실습 코드 에러 질문드립니다.

0

96

3

추론 결과의 Precision(또는 mAP) 평가 방법

0

86

2

mmdetection mask rcnn inferenct 실습 시 runpod 템플릿 관해서 질문드립니다.

0

61

2

runpod에서 google drive 연결 시 오류 발생

0

110

2

로드맵 선택

0

67

1

mmcv

0

59

2

Anchor box의 Positive 처리 위치

0

63

2

해당 강의 runpod 적용 후 에러 제보드립니다

0

86

2

run pod credit 관련 제보

0

99

2

mmdetection 2.x과 3.x 호환 관련 표기

0

78

2

mm_faster_rcnn_train_kitti.ipynb 실행 오류

0

98

3

질문 드립니다.

0

80

3

mm_faster_rcnn_train_coco_bccd 실행 오류 질문드립니다.

0

79

1

강사님께 수정을 제안드리고 싶은 것이 있습니다.

0

94

1

google automl efficientdet 다운로드 및 설치 오류

0

74

1

이상 탐지에 사용할 비전 기술 조언 부탁드립니다.

0

103

2

OpenCV 관련 질문드립니다.

0

71

2

mmcv 설치관련해서 문의드려요

0

331

3

강의 구성 관련해서 질문이 있습니다

1

138

2

모델 변환 성능 질문드립니다.

0

122

1

NMS 로직 문의 드려요

0

114

2