Commit 6807ce27 authored by Guillaume HEUSCH's avatar Guillaume HEUSCH
Browse files

[architectures] removed unecessary import in DCGAN

parent 225fb644
#!/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),
......
......@@ -3,7 +3,6 @@
import torch
import torch.nn as nn
from .utils import weights_init
class DCGAN_generator(nn.Module):
......
Supports Markdown
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment