Open lixiangyussss opened 2 years ago
What is your run command?
If your GPU has no tensor core, you have to set half to False.
如果您的 GPU 没有张量核心,则必须将 half 设置为 False。
You are referring to when training, or in the test.py file, I set half_precision=False in my test.py.
你的运行命令是什么?
![Uploading image.png…](),i use python train.py
You should have the same mistake as me, please use train_aux.py for large models
@lixiangyussss Did you solve this problem? I am training YOLOv7 with custom dataset but it takes so much time and the convergence is extremely slow.
I encountered same issues as @sontung2310 . It took around 1 hour per epoch,and the evaluation mAP=0.293 after 143 epochs (still training slowly...). I used to train with same custom dataset using YOLOv5x,which took around 2 days completing 300 epochs with mAP ~ 0.4. I wonder if there are some settings got wrong.
i do not know why,in my dataset, V7 is not comparable with any model of YOLO series V3-V6. The convergence rate of V7 and V7-tiny I used is very slow over 300 epochs, only 0.1map after 100 rounds, and the accuracy will vary a lot even at the end of training,whether I use pretrained weights or not