Update Readme

README.md

@@ -26,7 +26,30 @@ This repo provides different pytorch implementation for training a deep learning
   ```
 
 ## Single-GPU implementation
+```
+usage: train_simple.py [-h] --run_name RUN_NAME [--random_seed RANDOM_SEED]
+                       [-et EPOCHS_PER_TEST] [-ep EPOCHS] [-bs BATCH_SIZE]
+                       [-w NUM_WORKERS] [--learning_rate LEARNING_RATE]
+                       [--weight_decay WEIGHT_DECAY] [--momentum MOMENTUM]
+                       [--gamma GAMMA]
 
+optional arguments:
+  -h, --help            show this help message and exit
+  --run_name RUN_NAME
+  --random_seed RANDOM_SEED
+  -et EPOCHS_PER_TEST, --epochs_per_test EPOCHS_PER_TEST
+                        Number of epochs per test/val
+  -ep EPOCHS, --epochs EPOCHS
+                        Total number of training epochs to perform.
+  -bs BATCH_SIZE, --batch_size BATCH_SIZE
+  -w NUM_WORKERS, --num_workers NUM_WORKERS
+  --learning_rate LEARNING_RATE
+                        The initial learning rate for SGD.
+  --weight_decay WEIGHT_DECAY
+                        Weight deay if we apply some.
+  --momentum MOMENTUM   Momentum value in SGD.
+  --gamma GAMMA         gamma value for MultiStepLR.
+```
 ## Multi-GPU implementation
 
 ## Pytorch-lightning implementation

pl-hydra/configs/train.yaml

@@ -68,4 +68,6 @@ train: True
 test: True
 
 # seed for random number generators in pytorch, numpy and python.random
-seed: 100
+seed: 100
+
+