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