hellozhuo/pidinet

对预训练模型继续训练时遇到问题

Opened this issue · 0 comments

liukaiming6563 commented

首先感谢您的工作！请问：我想在您提供的预训练模型“table5_pidinet.pth”上继续训练，但出现报错：optimizer.load_state_dict(checkpoint['optimizer'])
KeyError: 'optimizer'
请问您知道什么原因吗？谢谢！