KaihuaTang / ResNet50-Pytorch-Face-Recognition

Using Pytorch to implement a ResNet50 for Cross-Age Face Recognition
MIT License
137 stars 49 forks source link

version of pytorch and torchvision #1

Open axiniu opened 6 years ago

axiniu commented 6 years ago

@KaihuaTang ,Hi,When I run ' python main.py ' I get AttributeError: module 'torchvision.transforms' has no attribute 'Resize'.

KaihuaTang commented 6 years ago

As I remember, I used Pytorch v0.3.0 in this repository. According to this URL, you can use transforms.Scale, if you are working on a old version.