leeyeehoo / siamfc-pytorch-gpu-benchmark

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

(object 'conv1.1.num_batches_tracked' doesn't exist)

zzowenzz opened this issue · comments

Hi, when i am running the notebook, it seems that there is something wrong with function load_net(NET_PATH, siam) with "KeyError: "Unable to open object (object 'conv1.1.num_batches_tracked' doesn't exist)""

And I print the k in net.state_dict().items() and it shows
"conv1.0.weight
conv1.0.bias
conv1.1.weight
conv1.1.bias
conv1.1.running_mean
conv1.1.running_var
conv1.1.num_batches_tracked"
which has 1 "conv1.1.num_batches_tracked".

May I possibly know the problem of it?