YOLOv7

jmani 2023. 3. 14. 11:58

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>