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中的损失函数是归一化了的。

相关推荐
Apache IoTDB几秒前
天谋科技 CTO 乔嘉林:多模态时序数智化软件栈
人工智能·科技·iotdb·技术大会
咖啡星人k1 分钟前
2026 多智能体协作实战:把角色契约写进SPEC,MonkeyCode 云端跑通
人工智能·机器学习
正在走向自律1 分钟前
爆火全网的2026机器人运动会:从赛场竞速到具身智能产业化的技术全解析
人工智能·机器人·具身智能·人形机器人·机器人运动会·百米竞速
生活皆是风景3 分钟前
GEO玩明白,流量自动上门
大数据·人工智能·产品运营
B站计算机毕业设计超人4 分钟前
计算机毕业设计知识图谱(Neo4j)+大语言模型LLM+GraphRAG图检索增强技术的考研院校推荐、分数线预测与智能问答系统(源码+文档+PPT+讲解)
大数据·人工智能·语言模型·毕业设计·知识图谱·课程设计·推荐算法
海宇AI5 分钟前
零信任架构实战:基于海宇运营商近3个月平均账单构建自动化P2P信审网关
人工智能·架构·自动化·p2p
格林威5 分钟前
C# 图像异步落盘存储:基于Channel 配合 ArrayPool 实现异步落盘
开发语言·人工智能·数码相机·机器学习·计算机视觉·c#·视觉检测
u86887 分钟前
常发携手上海脉信落地电话客服智能体,解决客服进线痛点
大数据·人工智能
鲜于言悠9057 分钟前
OpenSpec+Superpowers实战:AI驱动SDD+TDD完整开发工作流
人工智能
m0_614523558 分钟前
故障排查:移动物体表面贴图为什么会漂移?从稳定纹理到连续帧验收
网络·人工智能·贴图