diff --git a/README.md b/README.md index cd9a25e878..bb9df123ea 100644 --- a/README.md +++ b/README.md @@ -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. diff --git a/examples/new_project_templates/trainer_gpu_cluster_template.py b/examples/new_project_templates/trainer_gpu_cluster_template.py index bbf05c5f45..8531ba4b13 100644 --- a/examples/new_project_templates/trainer_gpu_cluster_template.py +++ b/examples/new_project_templates/trainer_gpu_cluster_template.py @@ -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 }