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

[architectures] finished original DR-GAN

parent 4fb98320
No related branches found
No related tags found
No related merge requests found
......@@ -321,7 +321,7 @@ class DRGANOriginal_discriminator(nn.Module):
# ------------------------------------------
# --- average pool
nn.AvgPool2d(6, stride=1)
nn.AvgPool2d(6, stride=1),
# --- fully connected
nn.Linear(320, (number_of_ids + conditional_dim + 1))
......
......@@ -11,6 +11,10 @@ from .DRGAN import DRGAN_encoder
from .DRGAN import DRGAN_decoder
from .DRGAN import DRGAN_discriminator
from .DRGANOriginal import DRGANOriginal_encoder
from .DRGANOriginal import DRGANOriginal_decoder
from .DRGANOriginal import DRGANOriginal_discriminator
from .DCGAN import weights_init
# gets sphinx autodoc done right - don't remove it
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment