[trainers] added ConditionalGANTrainer in __init__, fixed display of error...

[trainers] added ConditionalGANTrainer in __init__, fixed display of error (tensor.item() for tensors with one element)

bob/learn/pytorch/trainers/ConditionalGANTrainer.py

@@ -186,7 +186,7 @@ class ConditionalGANTrainer(object):
         optimizerG.step()
 
         end = time.time()
-        logger.info("[{}/{}][{}/{}] => Loss D = {} -- Loss G = {} (time spent: {})".format(epoch, n_epochs, i, len(dataloader), errD.data[0], errG.data[0], (end-start)))
+        logger.info("[{}/{}][{}/{}] => Loss D = {} -- Loss G = {} (time spent: {})".format(epoch, n_epochs, i, len(dataloader), errD.item(), errG.item(), (end-start)))
         
       # save generated images at every epoch
       # TODO: model moved to CPU and back and I don't really know why (expected CPU tensor error)

bob/learn/pytorch/trainers/__init__.py

 from .CNNTrainer import CNNTrainer
 from .DCGANTrainer import DCGANTrainer
+from .ConditionalGANTrainer import ConditionalGANTrainer
 
 # gets sphinx autodoc done right - don't remove it
 __all__ = [_ for _ in dir() if not _.startswith('_')]