2024年7月18日
pytorch 自定义 dataloader 维度不对齐+广播机制导致不易察觉 bug
摘要： 很简单，自定义了一个 dataloader，出现以下不易察觉 bug inputs 维度：[bs, 4]，这个没问题 labels 维度：正确应该是 [bs, 1]，但是 dataloader 出来是 [bs] 模型的 outputs 维度：[bs， 1] 如果用 torch.mean(torch. 阅读全文
posted @ 2024-07-18 15:22 Cold_Chair 阅读(1) 评论(0) 推荐(0) 编辑