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

changed read me

Browse Source
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
Download Patch File Download Diff File Expand all files Collapse all files
examples/new_project_templates/trainer_gpu_cluster_template.py
+2 -2
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
}