Skip to content
Snippets Groups Projects
Commit 72e6d8b0 authored by lukas leufen's avatar lukas leufen
Browse files

new attribute model name for model class

parent cd3de9ce
No related branches found
No related tags found
2 merge requests!50release for v0.7.0,!38Lukas issue043 refac clearify use of trainable option
Show whitespace changes
Inline Side-by-side
src/model_modules/model_class.py
+ 1
− 0
View file @ 72e6d8b0
...@@ -27,6 +27,7 @@ class AbstractModelClass(ABC): ...@@ -27,6 +27,7 @@ class AbstractModelClass(ABC):
self.__model = None self.__model = None
self.__loss = None self.__loss = None
self.model_name = self.__class__.__name__
def __getattr__(self, name: str) -> Any: def __getattr__(self, name: str) -> Any:
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment