[script] fixed the DR-GAN training script - added the trainer

bob/learn/pytorch/scripts/train_drgan_multipie.py

@@ -65,6 +65,8 @@ from bob.learn.pytorch.architectures import DRGAN_encoder
 from bob.learn.pytorch.architectures import DRGAN_decoder
 from bob.learn.pytorch.architectures import DRGAN_discriminator
 
+from bob.learn.pytorch.trainers import DRGANTrainer
+
 
 def main(user_input=None):
   
@@ -114,7 +116,7 @@ def main(user_input=None):
   # WARNING with the transforms ... act on labels too, at some point, I may have to write my own
   # Also, in 'ToTensor', there is a reshape performed from: HxWxC to CxHxW
   face_dataset = MultiPIEDataset(root_dir='/idiap/temp/heusch/data/multipie-cropped-64x64', 
-                                 frontal_only=True, 
+                                 frontal_only=False, 
                                  transform=transforms.Compose([
                                    RollChannels(), # bob to skimage:  
                                    ToTensor(),
@@ -152,5 +154,6 @@ def main(user_input=None):
   # ===============
   # === TRAINER ===
   # ===============
-  #trainer = DRGANTrainer(encoder, decoder, discriminator, batch_size=batch_size, latent_dim=latent_dim, noise_dim=noise_dim, use_gpu=use_gpu, verbosity_level=verbosity_level)
-  #trainer.train(dataloader, n_epochs=epochs, output_dir=output_dir)
+  trainer = DRGANTrainer(encoder, decoder, discriminator, image_size, batch_size=batch_size,
+                         noise_dim=noise_dim, conditional_dim=conditional_dim, latent_dim=latent_dim, use_gpu=use_gpu, verbosity_level=verbosity_level)
+  trainer.train(dataloader, n_epochs=epochs, output_dir=output_dir)