qq_38619449的博客

这个报错出现在计算loss的时候，代码原先是正常的。我尝试了很多方法，我把cuda换成在cpu上跑，就正常运行了。原因是显存不够造成的，可以尝试把batch改小或者imagesize改小。

使用pytorch进行模型训练时，我们通常会将数据包装成dataLoader，然后放入模型进行训练。dataLoader实质上就是一个数据加载器，必须要传入一个数据集类dataset。pytorch提供的dataset是一个抽象类，需要自己实现该类。继承dataset需要实现getitem和len两个方法，具体如下:class MyDataSet(Dataset): def __init__(self, imgs, labels): self._data = imgs