Skip to content

Key error 10 while running the training code #49

Description

@Kamin90

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 

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions