您好,欢迎来到12图资源库!分享精神,快乐你我!我们只是素材的搬运工!!
  • 首 页
  • 当前位置:首页 > 开发 > WEB开发 >
    复杂运用PyTorch搭建GAN模型(2)
    时间:2021-08-25 21:17 来源:网络整理 作者:网络 浏览:收藏 挑错 推荐 打印

            noise = (torch.rand(real_inputs.shape[0], 128) - 0.5) / 0.5 

            noise = noise.to(device) 

            fake_inputs = G(noise) 

            fake_outputs = D(fake_inputs) 

            fake_label = torch.zeros(fake_inputs.shape[0], 1).to(device) 

     

            outputs = torch.cat((real_outputs, fake_outputs), 0

            targets = torch.cat((real_label, fake_label), 0

     

            D_loss = loss(outputs, targets) 

            D_optimizer.zero_grad() 

            D_loss.backward() 

            D_optimizer.step() 

     

            # Training the generator 

    (责任编辑:admin)