Skip to content
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

Migrate to pytorch 1.1.0 #36

Open
jessica-santos opened this issue Jun 19, 2019 · 3 comments
Open

Migrate to pytorch 1.1.0 #36

jessica-santos opened this issue Jun 19, 2019 · 3 comments

Comments

@jessica-santos
Copy link

Hi, the code will be update for a new version of pytorch?

I'm trying to do this by my own, but I'm new to pytorch and finding some issues. After changing the .data to item, I try to run the code but receive RuntimeError: cuDNN error: CUDNN_STATUS_BAD_PARAM

@yl-1993
Copy link

yl-1993 commented Jul 2, 2019

@jessica-santos change .data to .data.item() may work for you

@tueboesen
Copy link

You need to change things in both main.py and losses.py

I looked in the forks of the code and found someone who had already done the changes, and just copied those changes, I used this one:
https://github.com/bl0/mean-teacher/tree/master/pytorch/mean_teacher

@jessica-santos
Copy link
Author

Thank you!

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

3 participants