Update README.md

README.md

@@ -22,4 +22,4 @@ The project implementation supports the following:
 
 ## **Run the model**
 
-1. Run ```python train_model.py --log_every_n_steps=5 --backbone_model=fcn_resnet50 --batch_size=60 --gpus=1 --learning_rate=2.1029136274973522e-05 --max_epochs=100 --optimizer=Adam --local_mode=True --dataset_path=/.../resources/datasets/ice-tiles-dataset-badlight/256x256```. Change the number of GPUs for distributed learning support (the default is one). Change the dataset_path to your own file path (you can use the ```pwd``` command inside of the 256x256 directory to find it). If you encounter ```RuntimeError: CUDA out of memory```, please decrease the --batch_size.
+1. Run ```python train_model.py --log_every_n_steps=5 --backbone_model=fcn_resnet50 --batch_size=60 --gpus=1 --learning_rate=2.1029136274973522e-05 --max_epochs=100 --optimizer=Adam --local_mode=True --dataset_path=/.../resources/datasets/ice-tiles-dataset-badlight/256x256```. Change the number of GPUs for distributed learning support (the default is one). Change the dataset_path to your own file path (you can use the ```pwd``` command inside of the 256x256 directory to find it). If you encounter ```RuntimeError: CUDA out of memory```, please decrease the batch size.