其他分享
首页 > 其他分享> > 关于Pytorch中model.train()

关于Pytorch中model.train()

作者:互联网

model.train()将模型设置为训练状态,作用:使Dropout,batchnorm知道后有不同表现(具体参考Dropout,batchnorm源码),只有这两个关心True or False。
将模型设置为测试状态有两种方法:
1.model.train(mode=False)
2.model.eval() 相当于第一种方法

model.train()源码:

model.eval() 源码:

标签:False,train,Pytorch,源码,eval,model,batchnorm
来源: https://www.cnblogs.com/wmc258/p/14520098.html