Skip to content

Commit 1db2dc1

Browse files
author
Anthony Smith
committed
Fixed "--help" output in README.md
Updated arguments to match what main.py is looking for. Fixed incorrectly listed defaults, removed duplicate "--save-model", copied "--save_model" description from main.py.
1 parent 1bef748 commit 1db2dc1

File tree

1 file changed

+8
-9
lines changed

1 file changed

+8
-9
lines changed

mnist_hogwild/README.md

+8-9
Original file line numberDiff line numberDiff line change
@@ -10,17 +10,16 @@ The main.py script accepts the following arguments:
1010
```bash
1111
optional arguments:
1212
-h, --help show this help message and exit
13-
--batch_size input batch_size for training (default:64)
14-
--testing_batch_size input batch size for testing (default: 1000)
15-
--epochs EPOCHS number of epochs to train (default: 1000)
16-
--lr LR learning rate (default: 0.03)
13+
--batch-size input batch_size for training (default: 64)
14+
--test-batch-size input batch size for testing (default: 1000)
15+
--epochs EPOCHS number of epochs to train (default: 10)
16+
--lr LR learning rate (default: 0.01)
1717
--momentum SGD momentum (default: 0.5)
1818
--seed SEED random seed (default: 1)
19-
--mps enables macos GPU training
20-
--save_model For saving the current Model
21-
--log_interval how many batches to wait before logging training status
22-
--num_process how many training processes to use (default: 2)
19+
--save_model save the trained model to state_dict
20+
--log-interval how many batches to wait before logging training status (default: 10)
21+
--num-processes how many training processes to use (default: 2)
2322
--cuda enables CUDA training
23+
--mps enables macos GPU training
2424
--dry-run quickly check a single pass
25-
--save-model For Saving the current Model
2625
```

0 commit comments

Comments
 (0)