About build model - Githubissues

WuJie1010 / Facial-Expression-Recognition.Pytorch

A CNN based pytorch implementation on facial expression recognition (FER2013 and CK+), achieving 73.112% (state-of-the-art) in FER2013 and 94.64% in CK+ dataset

MIT License

1.77k stars 546 forks source link

About build model #142

Open roroson opened 8 months ago

roroson commented 8 months ago

I see you build the resnet18 model, why not use the resnet18 model torchversion.model provided and just modify the linear layer. Thank you so much!