PyTorch踩坑记录1

1 model.eval()无法关闭dropout

因为model.eval()控制self.training参数,只有用nn.Dropout(0.5)声明才能在调用model.eval()后关闭,用F.dropout(x, p=0.5)是没办法自动关闭的,需要手动把self.training的参数传入到F.dropout()里才行。

网上查到是因为model.eval()会影响继承nn.module类的self.方法的训练和测试,但是F.dropout更像是一个函数没继承nn,module。

另外,回归模型在后面几层应该避免使用dropout,最多在浅层使用。

2 BCEloss归一化

在tf2中BCEloss没有归一化,但是在torch中的损失函数是归一化了的。

相关推荐
俊哥V几秒前
AI 今日研究简报 · 2026-09-08
人工智能·ai
一切皆是因缘际会4 分钟前
具身智能
人工智能·microsoft·生活
微三云生态系统架构师-彭丹6 分钟前
区域拆分独立分红池系统设计:解决大盘分红迟滞的四维拆分引擎
大数据·人工智能
2601_9622982710 分钟前
【自动化测试】基于Selenium + Python的web自动化框架
自动化测试·python·selenium·测试用例·web框架
吴佳浩 Alben13 分钟前
构建企业级 DevOps 排错 Agent:从日志告警到自动化修复 PR
大数据·人工智能·语言模型·架构·自动化·ai编程·devops
2601_9620775313 分钟前
Python 条件表达式
python·语法·列表推导式·条件表达式·pep308
陈童学哦14 分钟前
DeepSeek Harness实测:6个玩法,写完代码自动跑自动改
人工智能
HyperAI超神经19 分钟前
机器人运动误差降低90.4%,英伟达/哈佛等提出Hydra-0,用Action Flow统一世界建模与控制
人工智能·深度学习·机器人·英伟达·世界模型
深圳元器猫21 分钟前
深度解析:霍尼韦尔压力传感器如何保障AI数据中心液冷系统安全
人工智能·元器猫·霍尼韦尔压力传感器