[script] added epoch and iteration number on the figure title

bob/learn/pytorch/scripts/show_training_images.py

@@ -66,6 +66,7 @@ def main(user_input=None):
       filename = os.path.join(data_dir, f)
       temp = filename.split('_')
       epoch = int(temp[-2])
+      iteration = int(temp[-1].split('.')[-2])
       if epoch < start_epoch:
         continue
 
@@ -79,7 +80,7 @@ def main(user_input=None):
 
       from matplotlib import pyplot
       fig, axarr = pyplot.subplots(1,2)
-      fig.suptitle("ID = {}".format(id_example))
+      fig.suptitle("Epoch {}, iteration {} => ID {}".format(epoch, iteration, id_example))
       axarr[0].set_title("Real pose = {}".format(real_pose))
       axarr[0].imshow(numpy.rollaxis(numpy.rollaxis(real_example, 2),2))
       axarr[1].set_title("Target pose = {}".format(target_pose))