-
Notifications
You must be signed in to change notification settings - Fork 47
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
demo.py #6
Comments
I also have that issue after I update pytorch 0.3 to pytorch 0.4. Who anybody solve this problem? |
I ran into the same issue. The underlying reason is that pytorch has changed its batchnorm some time between versions 0.2 and 0.4: pytorch/pytorch#8481 |
I also have the same issue and yes, the underlying reason is pytorch/pytorch#8481, however, downgrading is not a practical solution in long run. Anybody plan to address this? |
I found the solution: instead to downgrade pytorch to 0.2, you can filter the keys by ignoring
|
Traceback (most recent call last):
File "/home/weakalign-master/demo.py", line 77, in
model.FeatureExtraction.state_dict()[name].copy_(checkpoint['state_dict']['FeatureExtraction.' + name])
KeyError: 'FeatureExtraction.model.1.num_batches_tracked'
I tried your code, but there was KeyError like above
OrderedDict "checkpoint['state_dict']['FeatureExtraction.model.1.num_batches_tracked']" does not exist
I'd appreciate it if you could check for the error.
The text was updated successfully, but these errors were encountered: