diff --git a/src/model_modules/model_class.py b/src/model_modules/model_class.py
index bb4801acfb5c3a643aecbcfad9cfdb758258d0ef..1a5dd9da38520d6d732253015e8a67325e24c460 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: