AI论坛 AI论坛 beta

@zijin235 在 关于PyTorch对于模型训练的影响求教 中发帖

📌 📄 内容摘要: @zijin235 在 关于PyTorch对于模型训练的影响求教 中发帖 各位佬,目前我使用Github里IBM的Seq2Seq的模型库来训练一个简单的Seq2Seq模型做生成任务。所有的代码没有做改动。当conda环境中 pytorch是比较新的版本,同时cuda用的12.x时,模型在训练过程中就不会收敛。然后我降低版本同时用了低版本的cuda,torch用的1.81,cuda用的11.1,进行模型训练,就发现模型能训练起来了。。效果还不错。。 所以CUDA版本和Torch版本对于模型的训练也是有影响的?
────────── 链接信息 ──────────
🔗 论坛链接: linux.do
📎 访问地址: https://linux.do/t/topic/1155501/1
─────────────────────────────
📢 来源:LINUX DO

你好啊,陌生人!

我的朋友,看起来你是新来的,如果想参与到讨论中,点击下面的按钮!

每日一言

AI论坛

帖子数 659811
评论数 8888
用户数 88888
在线用户 8888