changed read me

This commit is contained in:
William Falcon 2019-06-28 13:51:28 -04:00
parent f4030a7cf8
commit 93fcd861c4
2 changed files with 4 additions and 4 deletions

View File

@ -151,10 +151,10 @@ pip install pytorch-lightning
# clone lightning for the demo
git clone https://github.com/williamFalcon/pytorch-lightning.git
cd pytorch-lightning/docs/source/examples
cd examples/new_project_templates/
# run demo (on cpu)
python fully_featured_trainer.py
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.

View File

@ -17,11 +17,11 @@ np.random.seed(SEED)
# ---------------------
# DEFINE MODEL HERE
# ---------------------
from docs.source.examples.example_model import ExampleModel
from .lightning_module_template import LightningTemplateModel
# ---------------------
AVAILABLE_MODELS = {
'model_template': ExampleModel
'model_template': LightningTemplateModel
}