[architectures] finished original DR-GAN

bob/learn/pytorch/architectures/DRGANOriginal.py

@@ -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))

bob/learn/pytorch/architectures/__init__.py

@@ -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