]> Repos - ml-audio-enhancer/commitdiff
Update README.md
authorAndrew Gundersen <53452154+gundersena@users.noreply.github.com>
Mon, 17 Feb 2020 23:27:58 +0000 (17:27 -0600)
committerGitHub <noreply@github.com>
Mon, 17 Feb 2020 23:27:58 +0000 (17:27 -0600)
README.md

index 8d00c3a83510355190d6b5a526d030386e5fc9eb..8fedb28fb660e0fae4bc178f0e643d1cce5ef0f2 100644 (file)
--- a/README.md
+++ b/README.md
@@ -12,5 +12,31 @@ Repository Structure
 * `./samples`: audio files and spectrograms generated from the neural net
 
 ```
+cd ./main
 python 3 main.py -h
 ```
+
+It will then ask you if you'd like to train or evaluate.  To get help for each respectivly, just type:
+
+```
+python 3 main.py train -h
+python 3 main.py eval -h
+```
+
+Argument Reference:
+
+'''
+Arguments:
+  -h, --help        help
+  -i MODEL_ID       an index you can add to querey the model later
+  -c FROM_CKPT      bool, begin training session from a checkpoint
+  -k NEW_DATA       bool, make new set of data
+  -d DIM_SIZE       size of each audio sample
+  -x NUM_FILES      number of files to train
+  -e EPOCHS
+  -b BATCH_SIZE
+  -o CYCLE_LENGTH   length of cycle in epochs (only for SGDR)
+  -m MAX_LR         max learning rate (only for SGDR)
+  -n MIN_LR         min learning rate (only for SGDR)
+```
+