We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
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
您好,我在运行您的代码的时候,出现了gpu_mem在每一次Epoch后都上涨的问题,最终导致显存爆了,请问这个是什么原因?
Epoch gpu_mem GIoU obj cls total targets img_size 0/99 1.84G 0.07251 0.03948 0.04516 0.1572 7 1024: 100%|██████████| 3938/3938 [17:07<00:00, 3.83it/s] Class Images Targets P R mAP@.5 mAP@.5:.95: 100%|██████████| 1325/1325 [03:46<00:00, 5.86it/s] all 5.3e+03 5.98e+04 0.175 0.335 0.278 0.124 Epoch gpu_mem GIoU obj cls total targets img_size 1/99 5.69G 0.05713 0.03359 0.01962 0.1103 0 1024: 100%|██████████| 3938/3938 [16:54<00:00, 3.88it/s] Class Images Targets P R mAP@.5 mAP@.5:.95: 100%|██████████| 1325/1325 [04:04<00:00, 5.42it/s] all 5.3e+03 5.98e+04 0.335 0.489 0.414 0.214 Epoch gpu_mem GIoU obj cls total targets img_size 2/99 6.63G 0.0563 0.03319 0.01625 0.1057 23 1024: 100%|█████████▉| 3937/3938 [16:50<00:00, 3.90it/s]
Traceback (most recent call last):
The text was updated successfully, but these errors were encountered:
自己解决了,其实就是8G显存太低了,batch size设置为1都爆了
Sorry, something went wrong.
请问是如何解决的?
No branches or pull requests
您好,我在运行您的代码的时候,出现了gpu_mem在每一次Epoch后都上涨的问题,最终导致显存爆了,请问这个是什么原因?
Traceback (most recent call last):
The text was updated successfully, but these errors were encountered: