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.

Sign up for GitHub

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

Jump to bottom

Change view to reshape #6

Open

sdamadi opened this issue Dec 11, 2020 · 0 comments

sdamadi commented Dec 11, 2020

distributed-pytorch/imagenet_ddp_apex.py

Line 557 in b2a3e17

correct_k = correct[:k].view(-1).float().sum(0, keepdim=True)

Please change view to reshape to be compatible with Pytorch >1.6

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment