Module.train Module.eval 方法和 Module.training属性的关系

weixin_36049506

于 2019-05-22 10:44:46 发布

阅读量1k

点赞数

CC 4.0 BY-SA版权

分类专栏： pytorch

本文链接：https://blog.youkuaiyun.com/weixin_36049506/article/details/90439378

pytorch 专栏收录该内容

6 篇文章

订阅专栏

博客聚焦于Module.train()、Module.eval()方法和Module.training属性的关系，这些内容属于信息技术领域中编程相关知识，对于理解模块的训练和评估状态有重要意义。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

Module.train()、Module.eval() 方法和 Module.training属性的关系

print(net.training, net.submodel1.training)
net.train() # 将本层及子层的training设定为True
net.eval() # 将本层及子层的training设定为False
net.training = True # 注意，对module的设置仅仅影响本层，子module不受影响
net.training, net.submodel1.training