RuntimeError: mat1 and mat2 shapes cannot be multiplied

RuntimeError: mat1 and mat2 shapes cannot be multiplied

RuntimeError: mat1 and mat2 shapes cannot be multiplied

1 目标问题

在这里插入图片描述

2 原因和解决方法

  • 原因:最后一个batch的数据少于一个batchsize
  • 解决方式:设置dataloader中drop_last=True

以上就是全部内容

你可能感兴趣的:(pytorch,python,pytorch,深度学习)