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 分钟前
可视化+AI:Visual ALM 正在重新定义2026年的项目协作
人工智能·项目管理工具·项目管理软件·项目管理系统·项目管理平台
szephyr7 分钟前
腾讯云 ADP 智能体上线后响应变慢,先查 RAG 召回链路还是 Workflow 分支?诊断路径与排查清单
大数据·人工智能·腾讯云
郝学胜-神的一滴8 分钟前
力扣 692:巧用小顶堆高效求解前K个高频单词
java·数据结构·python·程序人生·算法·leetcode·职场和发展
栈溢出的浪漫9 分钟前
Python单元测试框架覆盖率-Coverage
python·单元测试·工具·覆盖率·coverage
问商十三载9 分钟前
RAG的适用边界在哪里?2026年价值与局限详解
大数据·前端·人工智能
淼澄研学11 分钟前
Python构建AI应用:从环境配置到FastAPI部署的5个实操步骤
人工智能·python·fastapi
1570925113414 分钟前
优先队列:从Java源码到实战
开发语言·python
找方案19 分钟前
AI医疗大模型获批三类证:AI诊疗正式纳入正规医疗体系
人工智能·dubbo·ai医疗·ai医疗大模型·ai诊疗
黎阳之光27 分钟前
黎阳之光:用“视频孪生”擦亮国门智慧底色
大数据·人工智能·物联网·安全·数字孪生
ONEYAC唯样27 分钟前
唯样-MLCC、电阻、电感集体升温,AI时代正在重塑被动元件格局
人工智能