Skip to content
Snippets Groups Projects
Commit c55ad3c7 authored by François  LAURENT's avatar François LAURENT
Browse files

bugfix

parent b895a3c0
Branches
Tags
No related merge requests found
...@@ -16,7 +16,7 @@ def train_model(backend, pretrained_model_instance="default"): ...@@ -16,7 +16,7 @@ def train_model(backend, pretrained_model_instance="default"):
nlabels = len(dataset.labels) nlabels = len(dataset.labels)
assert 0 < nlabels assert 0 < nlabels
# copy the pretrained model into the model instance directory # copy the pretrained model into the model instance directory
pretrained_autoencoder_dir = backend.model_dir() / "pretrained_models" / pretrained_model_instance pretrained_autoencoder_dir = backend.project_dir / "pretrained_models" / pretrained_model_instance
config_file = None config_file = None
for file in pretrained_autoencoder_dir.iterdir(): for file in pretrained_autoencoder_dir.iterdir():
if not file.is_file(): if not file.is_file():
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment