-
Notifications
You must be signed in to change notification settings - Fork 572
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
train all nan? #20
Comments
sync bn can not work with single gpu. could you show train_batch0.jpg in your runs/expxx folder? |
1.I can display rain_batch0.jpg correctly. Namespace(adam=False, batch_size=16, bucket='', cache_images=False, cfg='./models/yolov4-p5.yaml', data='./data/xx.yaml', device='1', epochs=450, evolve=False, global_rank=-1, hyp='data/hyp.finetune.yaml', img_size=[512, 512], local_rank=-1, logdir='runs/', multi_scale=False, name='yolov4-p5-tune', noautoanchor=False, nosave=False, notest=False, rect=False, resume='get_last', single_cls=False, sync_bn=False, total_batch_size=16, weights='yolov4-p5.pt', world_size=1)
0 -1 1 928 models.common.Conv [3, 32, 3, 1] Transferred 935/943 items from yolov4-p5.pt Analyzing anchors... anchors/target = 4.77, Best Possible Recall (BPR) = 0.9877
|
please provide the log file for training from scratch, not for fine-tuning. |
我采用如下命令从头训练得到了同样的结果: |
可以提供您在兩邊所使用的.data和.yaml嗎 |
@l1uw3n I'm having the same problem. Could you fix it? |
I'm running it like this: |
I use :python train.py --batch-size 16 --img 512 512 --data person.yaml --cfg yolov4-p5.yaml --weights yolov4-p5.pt --sync-bn --device 1 --name yolov4-p5-tune --hyp 'data/hyp.finetune.yaml' --epochs 450 --resume
result:
The text was updated successfully, but these errors were encountered: