search

pytorch / vision

Datasets, Transforms and Models specific to Computer Vision
https://pytorch.org/vision
BSD 3-Clause "New" or "Revised" License
16.33k stars 6.97k forks source link

How trained this code on data set fer2013? #1438

Closed akram-nejati closed 5 years ago

akram-nejati commented 5 years ago

I trained resnet on data set fer2013 with resnet in tensor flow, but how training in code on pytorch and resnet?

pmeier commented 5 years ago

If you are unfamiliar with PyTorch (torch), I suggest you start with the official tutorials to get a feel for it. Afterwards you can find the scripts that are used to train the models included in the torchvision package here:

https://github.com/pytorch/vision/tree/master/references

Since FER2013 is a classification dataset you can adopt

https://github.com/pytorch/vision/blob/master/references/classification/train.py

to fit your needs.