diff --git a/examples/new_project_templates/trainer_gpu_cluster_template.py b/examples/new_project_templates/trainer_gpu_cluster_template.py index 8531ba4b13..6aefd666e7 100644 --- a/examples/new_project_templates/trainer_gpu_cluster_template.py +++ b/examples/new_project_templates/trainer_gpu_cluster_template.py @@ -17,7 +17,7 @@ np.random.seed(SEED) # --------------------- # DEFINE MODEL HERE # --------------------- -from .lightning_module_template import LightningTemplateModel +from examples.new_project_templates.lightning_module_template import LightningTemplateModel # --------------------- AVAILABLE_MODELS = {