欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

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 即可

相关标签: GAN