diff --git a/examples/new_project_templates/multi_node_cluster_template.py b/examples/new_project_templates/multi_node_cluster_template.py index f8be3f27..f47d84f1 100644 --- a/examples/new_project_templates/multi_node_cluster_template.py +++ b/examples/new_project_templates/multi_node_cluster_template.py @@ -21,7 +21,7 @@ def main_local(hparams): main(hparams, None, None) -def main(hparams, cluster, results_dict): +def main(hparams, cluster): """ Main training routine specific for this project :param hparams: