diff --git a/examples/new_project_templates/multi_node_cluster_template.py b/examples/new_project_templates/multi_node_cluster_template.py index 2f38cd0d95..3e144259fd 100644 --- a/examples/new_project_templates/multi_node_cluster_template.py +++ b/examples/new_project_templates/multi_node_cluster_template.py @@ -20,11 +20,6 @@ np.random.seed(SEED) from lightning_module_template import LightningTemplateModel # --------------------- -AVAILABLE_MODELS = { - 'model_template': LightningTemplateModel -} - - """ Allows training by using command line arguments Run by: