在batch_size时,可能某些地方需要batch>1的时候才能运行,但在创建dataloader时,可能最后一个batch的大小正好为1,所以会导致这种错误。
为了解决这种情况,将dataloader的参数drop_last设置为True就可以了。
在batch_size时,可能某些地方需要batch>1的时候才能运行,但在创建dataloader时,可能最后一个batch的大小正好为1,所以会导致这种错误。
为了解决这种情况,将dataloader的参数drop_last设置为True就可以了。