KeyError：'optimizer_state_dict'

Question

我尝试在另一个数据集上使用预训练的模型，以使用相同的代码在我的数据集上恢复训练，但是下载模型并运行 main.py 到 laod 模型后出现错误

出现这个错误

Traceback (most recent call last):
  File "main.py", line 348, in <module>
    main()
  File "main.py", line 299, in main
    model, optimizer, epoch_idx, ckpt_dict = load_model(args.model_path, model, optimizer)
  File "/home/s44280162/Lipreading_using_Temporal_Convolutional_Networks3/lipreading/utils.py", line 170, in load_model
    optimizer.load_state_dict(checkpoint['optimizer_state_dict'])
KeyError: 'optimizer_state_dict'

这是 GitHub 链接，使用他们的代码和预先调整的模型。

https://github.com/mpc001/Lipreading_using_Temporal_Convolutional_Networks#model-zoo

Answer 1

此错误是由于 Python 字典中不存在键（在您的情况下，您的检查点没有任何名为“optimizer_state_dict”的键）。您可以使用 .keys() 方法打印字典键并检查检查点是否包含“optimizer_state_dict”。