使用PyTorch进行训练和测试时一定注意要把实例化的model指定train/eval

来源:互联网 发布:剑灵天族女捏脸数据韩 编辑:程序博客网 时间:2024/06/14 17:30

使用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()

阅读全文
0 0
原创粉丝点击