DEV Community · 2024-12-14T07:56:50Z 在使用DeepSpeed训练时遇到张量大小不匹配的问题 💡 原文英文，约200词，阅读约需1分钟。 📝 内容提要 在使用Hugging Face的Trainer和DeepSpeed集成训练模型时，遇到张量大小不匹配的错误，具体是张量a的大小（50）与张量b的大小（3）在非单例维度上不匹配。已检查输入ids和标签形状一致，确认批量大小，并确保模型在设备上正确放置。 🏷️ 标签 DeepSpeed Hugging Face Trainer 张量 张量大小 模型训练 阅读原文 生成长图 分享链接 已复制链接
