weiyangdaren

我遇到了和你类似的问题。我使用的是pytorch0.4.1和作者提供的模型finetune_300.tar。  然后我改成pytorch0.4.0，但是发生了错误“num_batches_tracked”。于是我将psmnet/sunmission.py里的 model.load_state_dict(state_dict['state_dict']) 改成 model.load_state_dict({k.replace('.num_batches_tracked', ''):v for k,v in state_dict['state_dict'].items()}) 它成功了  希望能帮到你