changed read me
This commit is contained in:
parent
f4030a7cf8
commit
93fcd861c4
|
@ -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.
|
||||
|
|
|
@ -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
|
||||
}
|
||||
|
||||
|
||||
|
|
Loading…
Reference in New Issue