检测网络测试时KeyError: 'session'

text后出现以下错误

Traceback (most recent call last):
  File "main.py", line 839, in <module>
    test()
  File "main.py", line 820, in test
    Network, optimizer = init_network(args, len(cls_list))
  File "main.py", line 158, in init_network
    args.session = checkpoint['session']
KeyError: 'session'

找了一大圈，感觉好像是权重文件的问题。我在其他地方下载的权重文件，直接进行text按照提示把文件名改成了要求的名字。运行就出现了这样的错误。有人说好像的checkpoint的名字不对，我不清楚怎么改。下载其他的权重文件检测有什么注意事项没有？