DeepDeblur-PyTorch
DeepDeblur-PyTorch copied to clipboard
Key error 10 while running the training code
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