pytorch lighting: Trying to resize storage that is not resizable

问题

在用pytorch lighting进行训练时碰到如下错误


即 Trying to resize storage that is not resizable 。

解决方案

在dataloader采样图片以及label时，保证每次采样的图片的分辨率不变。