Skip to content
Snippets Groups Projects
Commit 7ee698ae authored by leufen1's avatar leufen1
Browse files

model setup can now report better the layer configuration

parent 1fea8025
Branches
No related tags found
5 merge requests!430update recent developments,!413update release branch,!412Resolve "release v2.0.0",!406Lukas issue368 feat prepare cnn class for filter benchmarking,!403Resolve "prepare CNN class for filter benchmarking"
Pipeline #93710 passed
...@@ -202,5 +202,6 @@ class ModelSetup(RunEnvironment): ...@@ -202,5 +202,6 @@ class ModelSetup(RunEnvironment):
@staticmethod @staticmethod
def _clean_name(orig_name: str): def _clean_name(orig_name: str):
mod_name = re.sub(r'^{0}'.format(re.escape("<")), '', orig_name).replace("'", "").split(" ") mod_name = re.sub(r'^{0}'.format(re.escape("<")), '', orig_name).replace("'", "").split(" ")
mod_name = mod_name[1] if any(map(lambda x: x in mod_name[0], ["class", "function", "method"])) else mod_name[0] mod_name = mod_name[1] if any(map(lambda x: x in mod_name[0], ["class", "function", "method"])) else mod_name
return mod_name[:-1] if mod_name[-1] == ">" else mod_name mod_name = mod_name[0] if len(mod_name) == 1 else mod_name
return mod_name[:-1] if mod_name[-1] == ">" else "".join(mod_name)
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment