You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
The training code is stopped automatically at epoch 10 due to key error 10. I'm not exactly getting why this error occurs.
Could you help me to solve this error.
File "main.py", line 67, in <module>
main()
File "main.py", line 64, in main
main_worker(args.rank, args)
File "main.py", line 54, in main_worker
trainer.test(epoch)
File "/home/DeepDeblur-PyTorch-master/src/train.py", line 196, in test
self.evaluate(epoch, 'test')
File "/home/DeepDeblur-PyTorch-master/src/train.py", line 182, in evaluate
tq.set_description(self.criterion.get_loss_desc())
File "/home/DeepDeblur-PyTorch-master/src/loss/__init__.py", line 311, in geoss_desc
loss = self.loss_stat[self.mode]['Total'][self.epoch]
KeyError: 10
The text was updated successfully, but these errors were encountered:
The training code is stopped automatically at epoch 10 due to key error 10. I'm not exactly getting why this error occurs.
Could you help me to solve this error.
The text was updated successfully, but these errors were encountered: