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

bugfix

parent b895a3c0
No related branches found
No related tags found
No related merge requests found
......@@ -16,7 +16,7 @@ def train_model(backend, pretrained_model_instance="default"):
nlabels = len(dataset.labels)
assert 0 < nlabels
# 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
for file in pretrained_autoencoder_dir.iterdir():
if not file.is_file():
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment