diff --git a/_parameters.py b/_parameters.py
index 0806eed..3d4d30f 100644
--- a/_parameters.py
+++ b/_parameters.py
@@ -31,7 +31,7 @@ main_arg_parser.add_argument("--data_use_preprocessed", type=strtobool, default=
 # Training
 main_arg_parser.add_argument("--train_outpath", type=str, default="output", help="")
 main_arg_parser.add_argument("--train_version", type=strtobool, required=False, help="")
-main_arg_parser.add_argument("--train_epochs", type=int, default=500, help="")
+main_arg_parser.add_argument("--train_epochs", type=int, default=250, help="")
 main_arg_parser.add_argument("--train_batch_size", type=int, default=10, help="")
 main_arg_parser.add_argument("--train_lr", type=float, default=1e-3, help="")
 main_arg_parser.add_argument("--train_weight_decay", type=float, default=1e-8, help="")
diff --git a/multi_run.py b/multi_run.py
index f0ba3ce..05fe96c 100644
--- a/multi_run.py
+++ b/multi_run.py
@@ -1,26 +1,27 @@
 import warnings
 
-from _templates.new_project.utils.project_config import Config
-
 warnings.filterwarnings('ignore', category=FutureWarning)
 warnings.filterwarnings('ignore', category=UserWarning)
 
 # Imports
 # =============================================================================
 
-from _templates.new_project.main import run_lightning_loop, args
+from main import run_lightning_loop
+from utils.project_config import ThisConfig
+from _parameters import args
 
 
 if __name__ == '__main__':
 
     # Model Settings
-    config = Config().read_namespace(args)
+    config = ThisConfig().read_namespace(args)
     # bias, activation, model, norm, max_epochs
-    cnn_classifier = dict(train_epochs=10, model_use_bias=True, model_use_norm=True, data_batchsize=512)
+    pn2 = dict(model_type='PN2',model_use_bias=True, model_use_norm=True, data_batchsize=250)
+    p2g = dict(model_type='P2G', model_use_bias=True, model_use_norm=True, data_batchsize=250)
     # bias, activation, model, norm, max_epochs
 
-    for arg_dict in [cnn_classifier]:
-        for seed in range(5):
+    for arg_dict in [p2g]:
+        for seed in range(10):
             arg_dict.update(main_seed=seed)
 
             config = config.update(arg_dict)
diff --git a/utils/module_mixins.py b/utils/module_mixins.py
index 1337f5f..3806649 100644
--- a/utils/module_mixins.py
+++ b/utils/module_mixins.py
@@ -213,6 +213,8 @@ class BaseValMixin:
         disp.plot(include_values=True)
         self.logger.log_image('Confusion Matrix', image=disp.figure_, step=self.current_epoch)
 
+        plt.close('all')
+
         return summary_dict