Migrate to pytorch 1.1.0

CuriousAI / mean-teacher

A state-of-the-art semi-supervised method for image recognition

https://arxiv.org/abs/1703.01780

Other

1.56k stars 331 forks source link

Migrate to pytorch 1.1.0 #36

Open jessica-santos opened 5 years ago

jessica-santos commented 5 years ago

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 commented 5 years ago

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

tueboesen commented 4 years ago

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 commented 4 years ago

Thank you!