How to run on CPU? - 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.8k stars 553 forks source link

How to run on CPU? #101

Closed rathishkumar closed 4 years ago

rathishkumar commented 4 years ago

Getting below error on CPU machine:

RuntimeError: Attempting to deserialize object on a CUDA device but torch.cuda.is_available() is False. If you are running on a CPU-only machine, please use torch.load with map_location=torch.device('cpu') to map your storages to the CPU.

jinensetpal commented 4 years ago

on vizualize.py:

Comment each line of code that calls cuda [44 and 50]
Replace checkpoint = torch.load(os.path.join('FER2013_VGG19', 'PrivateTest_model.t7'), map_location=torch.device('cpu'))

jinensetpal commented 4 years ago

created a PR detailing updated code to this effect