From 72e6d8b0cd5894609cda0a846f62f8726affe507 Mon Sep 17 00:00:00 2001
From: lukas leufen <l.leufen@fz-juelich.de>
Date: Wed, 12 Feb 2020 13:15:32 +0100
Subject: [PATCH] new attribute model name for model class
---
src/model_modules/model_class.py | 1 +
1 file changed, 1 insertion(+)
diff --git a/src/model_modules/model_class.py b/src/model_modules/model_class.py
index bb4801ac..1a5dd9da 100644
--- a/src/model_modules/model_class.py
+++ b/src/model_modules/model_class.py
@@ -27,6 +27,7 @@ class AbstractModelClass(ABC):
self.__model = None
self.__loss = None
+ self.model_name = self.__class__.__name__
def __getattr__(self, name: str) -> Any:
--
GitLab