Skip to content

Commit db0dbf6

Browse files
authored
README.md : Add descriptions for optional arguments (pytorch#712)
Signed-off-by: nbqu <[email protected]>
1 parent dce9319 commit db0dbf6

File tree

1 file changed

+10
-0
lines changed

1 file changed

+10
-0
lines changed

vae/README.md

+10
Original file line numberDiff line numberDiff line change
@@ -7,3 +7,13 @@ It uses ReLUs and the adam optimizer, instead of sigmoids and adagrad. These cha
77
pip install -r requirements.txt
88
python main.py
99
```
10+
The main.py script accepts the following arguments:
11+
12+
```bash
13+
optional arguments:
14+
--batch-size input batch size for training (default: 128)
15+
--epochs number of epochs to train (default: 10)
16+
--no-cuda enables CUDA training
17+
--seed random seed (default: 1)
18+
--log-interval how many batches to wait before logging training status
19+
```

0 commit comments

Comments
 (0)