Skip to content
Snippets Groups Projects

Resolve "Add GANs"

Merged Guillaume HEUSCH requested to merge 4-add-gans into master
3 files
+ 0
5
Compare changes
  • Side-by-side
  • Inline
Files
3
#!/usr/bin/env python
# encoding: utf-8
import torch
import torch.nn as nn
class ConditionalGAN_generator(nn.Module):
""" Class implementating the conditional GAN generator
@@ -124,7 +121,6 @@ class ConditionalGAN_discriminator(nn.Module):
# input dimension
ndf = 64
self.main = nn.Sequential(
# input is (nc) x 64 x 64
nn.Conv2d((channels + conditional_dim), ndf, 4, 2, 1, bias=False),
Loading