使用PyTorch进行训练和测试时一定注意要把实例化的model指定train/eval
2017-11-03 15:11
429 查看
使用PyTorch进行训练和测试时一定注意要把实例化的model指定train/eval,eval()时,框架会自动把BN和DropOut固定住,不会取平均,而是用训练好的值,不然的话,一旦test的batch_size过小,很容易就会被BN层导致生成图片颜色失真极大!!!!!!
eg:
Class Inpaint_Network()
......
Model = Inpaint_Nerwoek()
#train:
Model.train(mode=True)
.....
#test:
Model.eval()
eg:
Class Inpaint_Network()
......
Model = Inpaint_Nerwoek()
#train:
Model.train(mode=True)
.....
#test:
Model.eval()
相关文章推荐
- 深度学习文章5:使用caffe对自己的图像数据进行训练并测试
- 使用Android Junit进行测试的注意事项
- Caffe学习:使用pycaffe进行网络训练与测试
- PyTorch(三)——使用训练好的模型测试自己图片
- eval解析JSON中的注意点 在JS中将JSON的字符串解析成JSON数据格式,一般有两种方式: 1.一种为使用eval()函数。 2. 使用Function对象来进行返回解析。 使用eval
- Pytorch打怪路(二)pytorch进行mnist训练和测试
- caffe学习笔记3:使用caffe对自己的图像数据进行训练和测试
- 使用opensmile提取音频的特征,得到特征向量,并扔进libsvm中进行分类训练测试
- 训练LSTM模型进行情感分类在IMDB数据集上,使用Keras API(Trains an LSTM model on the IMDB sentiment classification)
- 在使用Microsoft Visual Studio Test Controller 2010进行测试时,报如下错误: System.DllNotFoundException: 无法加载 DLL“Microsoft.VisualStudio.QualityTools.RecorderBarBHO100.dll”: 找不到指定的模块。 (异常来自 HRESULT:0x8007007E)。
- 使用Lenet5对mnist数据集进行训练和测试
- [人脸识别]使用VGG Face Model对一张图片进行测试
- sklearn.model_selection中的train_test_split使用方法,分割数据集,划分训练集和测试集
- Web前端——JQuery的进阶(测试使用 jQuery进行指定的操作)
- 使用回归树对美国波士顿房价训练数据进行学习,并对测试数据进行预测
- 使用conlleval.pl对CRF测试结果进行评价的方法
- 使用训练好的googlenet caffemodel进行图片分类
- 使用三种继承回归模型对美国波士顿房价训练数据进行学习,并对测试数据进行预测
- [Android学习笔记]使用Android Junit进行测试的注意事项
- 使用caffe的Python接口进行训练测试