UGATIT遇到【E-03】RuntimeError: element 0 of tensors does not require grad and does not have a grad

it2024-01-11  69

【E-03】RuntimeError: element 0 of tensors does not require grad and does not have a grad_fn

这个是因为没有将requires_grad设为True,Discriminator_loss = D_loss_A + D_loss_B中的Discriminator_loss默认是requires_grad为false,这个Discriminator_loss其实也是一个张量Tensor类型,将其的requires_grad改为True后,使用backward函数就可以得到requires_grad为True的所有参数的梯度。 下面再加一行Discriminator_loss = Discriminator_loss.requires_grad_()即可,下面的Generator_loss同理。

最新回复(0)