From 93fcd861c4eccee6df8dec0eac39b45ba7fa4c55 Mon Sep 17 00:00:00 2001 From: William Falcon Date: Fri, 28 Jun 2019 13:51:28 -0400 Subject: [PATCH] changed read me --- README.md | 4 ++-- .../new_project_templates/trainer_gpu_cluster_template.py | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) 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 }