报错内容:CUDA error: device-side assert triggered
CUDA error: device-side assert triggered
原因:使用ResNet50训练时使用了pretrained=True的模型,但实际类别数classes远超过MNIST自带的1000上限,导致标签越界。
pretrained=True
classes
解决方法:改成使用pretrained=False的模型,并指定num_classes
pretrained=False
num_classes
这是开发版吧?
写的简明扼要
支持,写的简单明了
背景还是太透明了点,没看几行眼睛就不舒服了,楼主把透明度再弄高一点吧
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】凌霞软件回馈社区,博客园 & 1Panel & Halo 联合会员上线
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步
· 本地部署 DeepSeek:小白也能轻松搞定!
· 传国玉玺易主,ai.com竟然跳转到国产AI
· 自己如何在本地电脑从零搭建DeepSeek!手把手教学,快来看看! (建议收藏)
· 我们是如何解决abp身上的几个痛点
· 如何基于DeepSeek开展AI项目