Skip to content
Snippets Groups Projects

Fix lr schedule

Merged Tim Tobias Bauerle requested to merge fix-lr-schedule into main
@@ -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):
Loading