Browse Source

README.md: updating multi-gpu syntax

master
Rafael Valle 6 years ago
parent
commit
2aba10cc0a
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      README.md

+ 1
- 1
README.md View File

@ -31,7 +31,7 @@ Distributed and FP16 support relies on work by Christian Sarofeen and NVIDIA's
2. (OPTIONAL) `tensorboard --logdir=outdir/logdir` 2. (OPTIONAL) `tensorboard --logdir=outdir/logdir`
## Multi-GPU (distributed) and FP16 Training ## Multi-GPU (distributed) and FP16 Training
1. `python -m multiproc train.py --output_directory=/outdir --log_directory=/logdir --hparams=distributed_run=True --fp16_run=True`
1. `python -m multiproc train.py --output_directory=outdir --log_directory=logdir --hparams=distributed_run=True,fp16_run=True`
## Inference ## Inference
1. `jupyter notebook --ip=127.0.0.1 --port=31337` 1. `jupyter notebook --ip=127.0.0.1 --port=31337`

Loading…
Cancel
Save