[trainer] fixed the epoch number in the logging

bob/learn/pytorch/trainers/CNNTrainer.py

@@ -172,5 +172,5 @@ class CNNTrainer(object):
           losses.append(loss.item())
       
       # do stuff - like saving models
-      logger.info("EPOCH {} DONE")
+      logger.info("EPOCH {} DONE".format(epoch+1))
       self.save_model(output_dir, epoch=(epoch+1), iteration=0, losses=losses)