model.eval()时测试结果远远低于model.train(),如何解决?

  1. 检查输入数据(train与test)是否经过了归一化
  2. 设置model.eval()时,网络中所有bn层里超参 track_running_stats = False
    但是,利用这种方式,模型在进行infer的时候必须要保持和训练时大致相当的batchsize,否则还是无法保持训练时的精度。
  3. 直接将BN层替换掉,尝试替换为layernorm或groupnorm。在我当时的任务中,将BN全部替换为layernorm会导致模型无法收敛;将BN层全部替换为groupnorm可以保证infer时精度与train时相当,且infer时batchsize可以为1,只是精度上相较于BN会降一些。
    另:替换为groupnorm时,参数设置为32
# 2. 中设置track_running_stats = False方法
def set_bn_fix_1(m):
    classname = m.__class__.__name__
    if classname.find('BatchNorm') != -1:
        m.track_running_stats = False
        
def test(data, model, criterion):
    print("******** Testing ********")
    with torch.no_grad():
        model.eval()
        #print("model eval===================")
        model.apply(set_bn_fix_1)
        #print(model)

  • 6
    点赞
  • 6
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值