咨询下，该报错怎么解决？

[zhanghaobucunzai]

../models/google/inception3_spg.py:137: UserWarning: nn.init.xavier_uniform is now deprecated in favor of nn.init.xavier_uniform_. nn.init.xavier_uniform(m.weight.data) => loading checkpoint '../models/inception_v3_google-1a9a5a14.pth' KeyError Traceback (most recent call last): File "../utils/Restore.py", line 24, in restore args.current_epoch = checkpoint['epoch'] + 1 KeyError: 'epoch'

During handling of the above exception, another exception occurred:

Traceback (most recent call last): File "train_frame.py", line 194, in train(args) File "train_frame.py", line 98, in train model, optimizer= get_model(args) File "train_frame.py", line 88, in get_model restore(args, model, optimizer, including_opt=False) File "../utils/Restore.py", line 38, in restore _model_load(model, checkpoint) File "../utils/Restore.py", line 51, in _model_load if model_dict.keys()[0].startswith('module.'): TypeError: 'odict_keys' object is not subscriptable

[xiaomengyc]

The checkpoint is not correctly loaded. Place the checkpoint at the right location or change the path to the checkpoint.