diff --git a/bob/ip/binseg/engine/trainer.py b/bob/ip/binseg/engine/trainer.py
index b225f69c995bc74cc2c3b2037a6969a6639f063c..06d55c164d13bf03c688b808c2c9f966ecd7ced5 100644
--- a/bob/ip/binseg/engine/trainer.py
+++ b/bob/ip/binseg/engine/trainer.py
@@ -114,9 +114,10 @@ def do_train(
 
             # progress bar only on interactive jobs
             for samples in tqdm(
-                data_loader, desc="batches", leave=False, disable=None
+                data_loader, desc="batches", leave=False, disable=None,
             ):
 
+                # data forwarding on the existing network
                 images = samples[1].to(device)
                 ground_truths = samples[2].to(device)
                 masks = None
@@ -125,6 +126,7 @@ def do_train(
 
                 outputs = model(images)
 
+                # loss evaluation and learning (backward step)
                 loss = criterion(outputs, ground_truths, masks)
                 optimizer.zero_grad()
                 loss.backward()