SeungjunNah/DeepDeblur-PyTorch

Key error 10 while running the training code

Kamin90 opened this issue · 0 comments

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