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

相关推荐
sendnews21 小时前
红松APP首秀北京老博会,“有温度的科技”赋能退休兴趣生活
人工智能·物联网
stayhungry_c21 小时前
Quarto生成PDF无法正常显示中文的问题
python·macos·r语言·pdf
美团技术团队21 小时前
ICCV 2025 | 美团论文精选及多模态推理竞赛冠军方法分享
人工智能
wwlsm_zql21 小时前
百度文心大模型再攀高峰:飞桨赋能AI,深度学习实力见证
人工智能·百度·paddlepaddle
程序员小远21 小时前
selenium元素定位---(元素点击交互异常)解决方法
自动化测试·软件测试·python·selenium·测试工具·测试用例·交互
王者鳜錸21 小时前
基于Selenium和AI的图像处理
图像处理·人工智能·selenium
ColderYY21 小时前
DrissionPage自动化
python·自动化
song1502653729821 小时前
全检垫圈垫片 视觉检测设备 在线自动化瑕疵检测机
人工智能·自动化·视觉检测
Python大数据分析@21 小时前
如何用 Python xlwings库自动化操作 Excel?
python·自动化·excel
song1502653729821 小时前
铜鼻子冷压端子视觉检测机 尺寸外观瑕疵自动化检测设备
人工智能·自动化·视觉检测