the version of pytorch - Githubissues

ShichenLiu / CondenseNet

CondenseNet: Light weighted CNN for mobile devices

MIT License

694 stars 131 forks source link

the version of pytorch #22

Open jqsun98 opened 5 years ago

jqsun98 commented 5 years ago

There are several problems of train-loader from torch.vision using my torch version (1.0.1), so could you show the requirements.txt or a latest code version using torch 1.0.1 ?

ShichenLiu commented 5 years ago

Hi, I have updated and tested the code to support PyTorch 1.1.0. Please feel free to check it out!