readme update

buildingamind · Dec 1, 2023 · 627c67a · 627c67a
1 parent 4686c6a
commit 627c67a
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 0 deletions.
diff --git a/README.md b/README.md
@@ -39,6 +39,16 @@ pip install -r requirements.txt
 
 ## Model Training
 
+The bash script 'train_vit.sh' is used to train the model. The bash script calls the python script which intializes the model and the dataloader.
+
+```python
+sh train_vit.sh
+```
+
+<p>Example</p>
+
+<img src="./media/bash.png">
+
 
 ## Model Testing
 

diff --git a/media/.DS_Store b/media/.DS_Store
diff --git a/media/bash.png b/media/bash.png