Falto implementar la función Disc_GAN_loss
def Disc_GAN_loss(D2, fake2, real2, fake_2_buffer, loss, target_real, target_fake):
pred_real = D2(real2)
loss_D2_real = loss(pred_real, target_real)
fake2 = fake_2_buffer.push_and_pop(fake2)
pred_fake = D2(fake2.detach())
loss_D2_fake = loss(pred_fake, target_fake)
loss_D2 = (loss_D2_real + loss_D2_fake) * 0.5
return loss_D2
La saqué del colab de JP -> https://github.com/JuanPabloMF/deep-learning-platzi/blob/master/DeepLearningConPytorch.ipynb
Saludos
¿Quieres ver más aportes, preguntas y respuestas de la comunidad? Crea una cuenta o inicia sesión.