Skip to content
Snippets Groups Projects
Commit 6807ce27 authored by Guillaume HEUSCH's avatar Guillaume HEUSCH
Browse files

[architectures] removed unecessary import in DCGAN

parent 225fb644
No related branches found
No related tags found
1 merge request!4Resolve "Add GANs"
#!/usr/bin/env python #!/usr/bin/env python
# encoding: utf-8 # encoding: utf-8
import torch import torch
import torch.nn as nn import torch.nn as nn
class ConditionalGAN_generator(nn.Module): class ConditionalGAN_generator(nn.Module):
""" Class implementating the conditional GAN generator """ Class implementating the conditional GAN generator
...@@ -124,7 +121,6 @@ class ConditionalGAN_discriminator(nn.Module): ...@@ -124,7 +121,6 @@ class ConditionalGAN_discriminator(nn.Module):
# input dimension # input dimension
ndf = 64 ndf = 64
self.main = nn.Sequential( self.main = nn.Sequential(
# input is (nc) x 64 x 64 # input is (nc) x 64 x 64
nn.Conv2d((channels + conditional_dim), ndf, 4, 2, 1, bias=False), nn.Conv2d((channels + conditional_dim), ndf, 4, 2, 1, bias=False),
......
...@@ -3,7 +3,6 @@ ...@@ -3,7 +3,6 @@
import torch import torch
import torch.nn as nn import torch.nn as nn
from .utils import weights_init
class DCGAN_generator(nn.Module): class DCGAN_generator(nn.Module):
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment