0389fda574d618ec22b488208a1d597a18d7ce72,examples/next-frame.py,NextFrameGAN,forward_pass,#NextFrameGAN#,195

Before Change

                if config.discriminator3d:
                    grems = grems[1:] + [rg[:,:,None,:,:]]
                    rc = rc[:,:,None,:,:]
                    d_fakes.append(D(torch.cat(grems, dim=2), context={"c":rc}))
                else:
                    grems = grems[1:] + [rg]
                    d_fakes.append(D(torch.cat(grems, dim=1), context={"c":rc}))

After Change

            d_fake = D(d_fake_input, context={"c": c})
            d_fakes.append(d_fake)
            _d_loss, _g_loss = loss.forward(d_real, d_fake)
            d_losses.append(_d_loss)
            g_losses.append(_g_loss)

        if len(rgs) > 0:
            grems = rgs[:len(rems)]

Instances