RuntimeError: shape '[64, 1, 1, 1]' is invalid for input of size 6400
程序员文章站
2022-03-09 13:26:43
...
pytorch报错RuntimeError: shape ‘[64, 1, 1, 1]’ is invalid for input of size 6400
源代码
samples = samples.data.reshape(batch_size, 1, 1, 1)
samples = samples.data.expand(batch_size, input_dim, 1, 1)
samples = samples.cuda() if use_cuda else samples
原因:
由于修改了resize -> reshape,导致第一次循环对samples的改变维持到了第二轮循环
修改
把后面的samples -> samples1 即可
上一篇: js如何实现select下拉框选择
下一篇: GAN教程