inflearn logo
강의

강의

N
챌린지

챌린지

멘토링

멘토링

N
클립

클립

로드맵

로드맵

지식공유

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

multi gpu사용방법

914

rlaxorud890

작성한 질문수 2

0

안녕하세요 강사님.
 
혹시 multi gpu사용을 위해서는 어떻게 설정을 해주어야하는지 알 수 있을까요???
 
 
찾아보니까 여기서 'find_unused_parameters'와 관련된 세팅을 해야하는 것 같은데 저 주석이 정확히 어떻게 하라는 건지 이해가 안되서 질문드립니다!
 
 
##추가질문
그리고 추가로 pretrained된 모델을 재학습시키는 것이 아닌 그냥 아직 학습이 하나도 되지 않은 모델을 학습시키려면 config에서 checkpoint을 통해서 pretrained된 모델을 불러오는 과정만 없애주면 되는건가요??

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

답변 1

0

권 철민

안녕하십니까, 

1. 저도 multi gpu를 적용해보진 않아선 정확히는 모르겠습니다. 

multi gpu를 적용하긴 위해서는 아래와 같이 dist_train.sh를 이용하셔야 합니다. dist_train.sh은 mmdetection github에 있습니다. https://github.com/open-mmlab/mmdetection 

./tools/dist_train.sh ${CONFIG_FILE} ${GPU_NUM} [optional arguments]

물론 dist_train.sh 역시 train_detector()를 사용하지만, 입력되는 인자등이 조금 다릅니다. 아래에서 train with multiple gpus 내용을 살펴 보시면 됩니다. 

https://mmdetection.readthedocs.io/en/v2.2.1/getting_started.html#train-a-model

2. 네 맞습니다. config에서 load_from = None 으로 설정해 주시면 됩니다.(Default가 None입니다)

 

0

rlaxorud890

감사합니다!

MMDetection 버전 이슈

0

50

2

강의 환경설정 질문

0

61

2

Custom Dataset에서의 polygon 정보 관련

0

111

3

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

0

97

2

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

0

120

3

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

0

94

2

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

0

68

2

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

0

125

2

로드맵 선택

0

74

1

mmcv

0

65

2

Anchor box의 Positive 처리 위치

0

70

2

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

0

95

2

run pod credit 관련 제보

0

123

2

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

0

88

2

mm_faster_rcnn_train_kitti.ipynb 실행 오류

0

114

3

질문 드립니다.

0

88

3

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

0

89

1

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

0

101

1

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

0

87

1

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

0

111

2

OpenCV 관련 질문드립니다.

0

86

2

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

0

355

3

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

1

141

2

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

0

125

1