RuntimeError: Mismatch in shape: grad_output[0] has a shape of torch.Size([1]) and output[0]...

最新推荐文章于 2024-06-20 21:40:33 发布

Prongs_

最新推荐文章于 2024-06-20 21:40:33 发布

阅读量1.4k

点赞数 3

文章标签： python 深度学习 pytorch GAN

本文链接：https://blog.csdn.net/Prongs_/article/details/129182947

版权

在训练tfvaegan模型时遇到了两个错误：第一个是关于形状不匹配的RuntimeError，涉及grad_output和output的形状；第二个是尝试从0维张量中获取索引的IndexError。通过将one初始化为torch.tensor并使用.item()方法来解决这些错误。修改后的代码成功避免了这两个问题，使模型训练得以继续。

摘要由CSDN通过智能技术生成

训练tfvaegan时出现以下错误：

Traceback (most recent call last):
  File "D:\Prongs Project\tfvaegan-master\train_images.py", line 172, in <module>
    criticD_real.backward(mone)
  File "D:\Anaconda\envs\pyt11\lib\site-packages\torch\_tensor.py", line 363, in backward
    torch.autograd.backward(self, gradient, retain_graph, create_graph, inputs=inputs)
  File "D:\Anaconda\envs\pyt11\lib\site-packages\torch\autograd\__init__.py", line 166, in backward
    grad_tensors_ = _make_grads(tensors, grad_tensors_, is_grads_batched=False)
  File