From cf63bd92bca6d72076d6d95e214fcb20eacfb506 Mon Sep 17 00:00:00 2001
From: vahidrezanezhad <vahid631983@gmail.com>
Date: Wed, 25 Jun 2025 18:33:55 +0200
Subject: [PATCH] Update README.md: how to train model using docker image

---
 README.md | 14 +++++++++++++-
 1 file changed, 13 insertions(+), 1 deletion(-)

diff --git a/README.md b/README.md
index b9e70a8..7c69a10 100644
--- a/README.md
+++ b/README.md
@@ -24,7 +24,19 @@ each class will be defined with a RGB value and beside images, a text file of cl
 
 ### Train
 To train a model, run: ``python train.py with config_params.json``
-      
+
+### Train using Docker
+
+#### Build the Docker image
+
+   ```bash
+   docker build -t model-training .
+   ```
+#### Run Docker image 
+   ```bash
+   docker run --gpus all -v /host/path/to/entry_point_dir:/entry_point_dir model-training
+   ```
+
 ### Ground truth format
 Lables for each pixel are identified by a number. So if you have a 
 binary case, ``n_classes`` should be set to ``2`` and labels should