feat(handlers): add common training handlers template (#26)

* feat(handlers): add common training handlers template

* extract each section into function, include template in main.py

* fix: add to_save, string quote, trainer

* chore: common training handlers default value

app/codegen.py

@@ -17,9 +17,11 @@ def __init__(self, templates_dir: str = "./templates", dist_dir: str = "./dist")
     def render_templates(self, template_name: str, config: dict):
         """Renders all the templates files from template folder for the given config."""
         self.rendered_code[template_name] = {}  # clean up the rendered code for given template
-        # loading the template files based on given template
+        # loading the template files based on given template and from the _base folder
+        # since we are using some templates from _base folder
+        loader = FileSystemLoader([self.templates_dir / "_base", self.templates_dir / template_name])
         env = Environment(
-            loader=FileSystemLoader(self.templates_dir / template_name),
+            loader=loader,
             trim_blocks=True,
             lstrip_blocks=True,
         )

templates/_base/_handlers.pyi

@@ -0,0 +1,19 @@
+    from ignite.contrib.engines import common
+    {% if setup_common_training_handlers %}
+    common.setup_common_training_handlers(
+        trainer=trainer,
+        train_sampler=None,
+        to_save=to_save,
+        output_path='{{ output_path }}',
+        save_every_iters={{ save_every_iters }},
+        n_saved={{ n_saved }},
+        log_every_iters={{ log_every_iters }},
+        with_pbars={{ with_pbars }},
+        with_pbar_on_iters={{ with_pbar_on_iters }},
+        stop_on_nan={{ stop_on_nan }},
+        clear_cuda_cache={{ clear_cuda_cache }},
+        lr_scheduler=None,
+        with_gpu_stats=False,
+        output_names=None,
+    )
+    {% endif %}