DeepDeblur-PyTorch icon indicating copy to clipboard operation
DeepDeblur-PyTorch copied to clipboard

Key error 10 while running the training code

Open Kamin90 opened this issue 2 years ago • 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 

Kamin90 avatar Oct 12 '22 04:10 Kamin90