(object 'conv1.1.num_batches_tracked' doesn't exist)
zzowenzz opened this issue · comments
zzowenzz commented
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?