深度学习预训练模型无法加载

在github上下载了用于人体姿态估计的堆叠沙漏网络，并且下载了作者提供的2HG的预训练模型
源码地址如下https://github.com/princeton-vl/pytorch_stacked_hourglass
但是在运行test.py时得到的结果如下

不知道是不是因为没有成功加载到预训练模型，如果想在test.py运行时载入预训练模型，应该怎么修改代码呢
现在我已经用作者的checkpoint.py替换了我自己训练的
(因为电脑显存问题自己训练的epoch为30，太小了，但自己训练完test时数值和这个也差不太多)

model = your_network
model.load_state_dict(torch.load(model_weight_path, map_location=device))
这只是个加载预训练权重的方法，具体的程序结合你的代码进行修改，CSDN你已开启陌生人防打扰，你虽然发消息给我，但无法私信联系。
如果只是为了跑通代码，不如结合已跑通的代码进行一行一行调试，不积跬步无以至千里！