Skip to content
Snippets Groups Projects
Commit b43bdda4 authored by Tim Tobias Bauerle's avatar Tim Tobias Bauerle
Browse files

Avoid common prefixes at model loading for testing

parent e0de9ea9
Branches
No related tags found
1 merge request!22Fix lr schedule
......@@ -43,7 +43,7 @@ class TestController(BaseController):
def _get_model_with_highest_postfix_number(self, model_save_path):
"""Returns the highest postfix number in the given directory for the configured model_prefix."""
model_prefix = self.__model_prefix__()
model_prefix = f"{self.__model_prefix__()}_client_" # assume server and client weights were saved appropriately
highest_postfix_number = 0
for file in os.listdir(model_save_path):
if file.startswith(model_prefix):
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment