mrq/DL-Art-School
1
3
Fork 5
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Add distributed training guide to docs

Browse Source
This commit is contained in:
James Betker 2021-06-06 16:56:40 -06:00
parent 2ad2b56438
commit 4dd053f694
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
README.md
View File

@ -77,9 +77,12 @@ DLAS comes with some Dataset instances that I have created for my own use. Unles
There are currently 3 base scripts for interacting with models. They all take a single parameter, `-opt` which specifies the configuration file which controls how they work. Configs (will be) documented above in the user guide.
#### train.py
Starts (or continues) a training session.
Start (or continue) a training session:
`python train.py -opt <your_config.yml>`
Start a distributed training session:
`python -m torch.distributed.launch --nproc_per_node=<gpus> --master_port=1234 train.py -o <opt> --launcher=pytorch`
#### test.py
Runs a model against a validation or test set of data and reports metrics (for now, just PSNR and a custom perceptual metric)
`python test.py -opt <your_config.yml>`