JMANI

YOLOv7 본문

AI/Neural Networks and Deep Learning

YOLOv7

jmani 2023. 3. 14. 11:58

train.py : command 옵션

529> --weights: 모델 가중치 파일, pre-trained weights 

530> --cfg: 

531> --

532>

533> --epochs: default 300. training 데이터셋을 학습시킨 횟수

534> --batch-size: default 16. 모든 GPU에서 총 배치 사이즈. training 데이터셋을 나누는 데이터 샘플 수

535> 

536>

537>

538> --nosave: 마지막 check point에서만 모델 저장

539> --notest: 학습 중에 validation 데이터 test X, 마지막 epoch에서만 test

540>

541>

542>

543>

544>

545> --device: default ''. i.e. 0 or 0,1,2,3 or cpu

546>

547>

548> --adam: torch.optim.Adam() optimizer 사용. default는 sgd

546> 

546>

546>

546>

546>

546>

546>

546>

546>

546>

546>

546>

train.py

 

'AI > Neural Networks and Deep Learning' 카테고리의 다른 글

Week 2: Neural Networks Basics  (0) 2021.10.26
1주차: Introduction to Deep Learning  (0) 2021.10.26
Comments