wassname/pytorch-lightning
1
0
Fork 0
mirror of https://github.com/wassname/pytorch-lightning.git synced 2026-09-10 12:21:57 +08:00
Code Issues Packages Projects Releases Wiki Activity

updated examples

Browse Source
This commit is contained in:
William Falcon
2019-07-25 12:30:18 -04:00
parent a1dd4d3e2c
commit 6bd58de40e
1 changed files with 9 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
README.md
+9 -5
View File
@@ -287,7 +287,7 @@ pip install pytorch-lightning
# clone lightning for the demo
git clone https://github.com/williamFalcon/pytorch-lightning.git
cd examples/new_project_templates/
cd pytorch_lightning/examples/new_project_templates/
# run demo (on cpu)
python trainer_gpu_cluster_template.py
@@ -295,11 +295,15 @@ python trainer_gpu_cluster_template.py
Without changing the model AT ALL, you can run the model on a single gpu, over multiple gpus, or over multiple nodes.
```bash
# run a grid search on two gpus
python fully_featured_trainer.py --gpus "0;1"
# train on cpu
python single_cpu_template.py
# train on multiple-gpus
python single_gpu_node_template.py --gpus "0,1"
# train on 32 gpus on a cluster
python multi_node_cluster_template.py --nb_gpu_nodes 4 --gpus '0,1,2,3,4,5,6,7'
# run single model on multiple gpus
python fully_featured_trainer.py --gpus "0;1" --interactive
```