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

相关推荐
思绪无限2 分钟前
YOLOv5至YOLOv12升级:血细胞检测系统的设计与实现(完整代码+界面+数据集项目)
人工智能·python·深度学习·目标检测·计算机视觉·yolov12·血细胞检测
碧海银沙音频科技研究院31 分钟前
杰理项目开发大全课程
人工智能·深度学习·算法
Change is good1 小时前
VS + Copilot 最简入门 AI Skill 模板
人工智能·copilot
Rubin智造社1 小时前
04月24日AI每日参考:GPT-5.5正式发布,DeepSeek获腾讯阿里争相入局
人工智能·claude code·deepseek v4·gpt-5.5·deepseek融资
imbackneverdie1 小时前
本科毕业论文自救指南
人工智能·ai·ai写作·毕业论文·ai工具·科研工具
生成论实验室1 小时前
生成态势猜想:一种统一的宇宙动力学语法
人工智能·科技·神经网络·算法·信息与通信
ZStack开发者社区1 小时前
DeepSeek-V4首发即支持,ZStack AIOS 私有化部署即刻可用
人工智能·开源·云计算
JEECG低代码平台1 小时前
四强同台!DeepSeek-V4-Pro / GPT-5.5 / GLM-5.1 / MiniMax M2.7 横评:到底该选谁?
人工智能·gpt
慧一居士1 小时前
Open WebUI 和 MaxKB 功能对比、使用场景对比、方案选型
人工智能
温迪的AI分享2 小时前
2026 AI抠图实测:4款工具效果与速度对比
图像处理·人工智能·经验分享·ai