diff --git a/content/tutorial/segmentation-training.md b/content/tutorial/segmentation-training.md
index 72e989ccbc4b88691c8f2e430a088806f6d5b5cd..8a3617e224aa7367bd20000f51d5d23a31e03ce3 100644
--- a/content/tutorial/segmentation-training.md
+++ b/content/tutorial/segmentation-training.md
@@ -124,14 +124,16 @@ Configure the `YOLO Training | Detect/Segment` worker by clicking on the button
 
 Select **New configuration** on the left column, to create a new configuration. Again, name it after the dataset you are using.
 
-{{ figure(image="tutorial/training/segmentation/train_configuration.png", height=500, caption="Worker configuration") }}
+{{ figure(image="tutorial/training/segmentation/train_configuration.png", height=1000, caption="Worker configuration") }}
 
 The most important parameters are:
 - *Model that will receive the new trained version*: search for the name of [your model](#create-a-model),
-- *Number of epochs[^epoch] to train[^training] the model*: the default value is good enough but you can set it to a larger number if you want to train for a longer period time,
+- *Class names to predict*: add the slugs of the types that the model will predict,
+- *Number of epochs[^epoch] to train[^training] the model*: the default value is good enough but you can set it to a larger number if you want to train for a longer period of time,
 - *Type of object to detect using the segmenter*: 
   - a segmenter will produce masks (polygons),
   - a detector will produce bounding boxes (rectangles).
+- *Batch size for training*: a higher value will make the training faster but will also increase the memory usage.
 
 Click on **Create** then **Save** when you are done filling the fields. Your process is ready to go.
 
diff --git a/content/tutorial/training/segmentation/train_configuration.png b/content/tutorial/training/segmentation/train_configuration.png
index 5a9a953d7e29400b8a14cea515311603a5c41e2b..1bdecce20b681b1cede846bd80f099bdd9008095 100644
Binary files a/content/tutorial/training/segmentation/train_configuration.png and b/content/tutorial/training/segmentation/train_configuration.png differ