Skip to content
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

after training completion of all epochs , its crashing, not able to get final map,best.pt weight #154

Open
akashAD98 opened this issue Dec 28, 2022 · 0 comments

Comments

@akashAD98
Copy link

akashAD98 commented Dec 28, 2022

i can able to do complete all training, once it reaches to last epoch its throwing an error & not able to get map,

also can we do testing of the trained model using detect.py?

the weight file size is 259 MB , I guess weight files are not downloading properly or they are corrupt.
i tested it with detect.py & getting

RuntimeError: PytorchStreamReader failed reading zip archive: failed finding central directory
  Epoch   gpu_mem       box       obj       cls    labels  img_size
   119/119     1.63G   0.02918   0.05866  0.005124         3       640: 100%|██████████| 2311/2311 [09:13<00:00,  4.18it/s]                                                                                
               Class     Images     Labels          P          R     mAP@.5 mAP@.5:.95: 100%|██████████| 548/548 [01:08<00:00,  7.96it/s]                                                                  
                 all       2192       8590      0.982      0.939      0.969      0.758
Traceback (most recent call last):
  File "train.py", line 695, in <module>
    main(opt)
  File "train.py", line 591, in main
    train(opt.hyp, opt, device, callbacks)
  File "train.py", line 449, in train
    torch.save(ckpt, best)
  File "/home/sciffer/.local/lib/python3.8/site-packages/torch/serialization.py", line 424, in save
    return
  File "/home/sciffer/.local/lib/python3.8/site-packages/torch/serialization.py", line 290, in __exit__
    self.file_like.write_end_of_file()
RuntimeError: [enforce fail at inline_container.cc:445] . PytorchStreamWriter failed writing central directory for archive best: file write failed

@iscyy @WangQvQ please let me know how can I solve this issue

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

1 participant