search

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.78k stars 545 forks source link

plot confusion matrix时被killed #35

Closed zionward closed 5 years ago

zionward commented 5 years ago

我是cpu-only的电脑。 在跑confusion matrix那一步（python plot_fer2013_confusion_matrix.py --model VGG19 --split PrivateTest）的时候，先报了RuntimeError（torch.cuda.is_available() is False）。我按照报错提示

WX20190505-155349@2x

将plot_fer2013_confusion_matrix.py中的checkpoint = torch.load(os.path.join(path, opt.split + '_model.t7')) 改为👉checkpoint = torch.load(os.path.join(path, opt.split + '_model.t7'), map_location='cpu') 然后按照之前issues的那样comment掉了.cuda()的code。 然后再run plotting confusion matrix的指令的时候并没有报错，但是会卡死，然后最后显示killed

WX20190505-155400@2x

可以帮忙看一下是什么问题吗？谢谢！

WuJie1010 commented 5 years ago

因为是因为一下子通过网络的参数量太大，cpu内存不够了，考虑在测试的时候调低batchsize尝试下。

zionward commented 5 years ago

哦，好的！谢谢！