RuntimeError: Expected hidden size (1, 3, 64), got [1, 5, 64]

错误的原因是训练集的大小TRAIN_SIZE = int(TRAIN_SIZE_PERCENT*total_len)

不能整除batch_size

RuntimeError: Expected hidden size (1, 3, 64), got [1, 5, 64]_第1张图片

使用pytorch时遇到的问题汇总_vivian_ll的博客-CSDN博客 

你可能感兴趣的:(Pytorch)