From c55ad3c747b30bbb0bbf0db120c78fedfc73b137 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Fran=C3=A7ois=20Laurent?= <francois.laurent@posteo.net>
Date: Mon, 1 Aug 2022 10:29:53 +0200
Subject: [PATCH] bugfix
---
src/maggotuba/models/train_model.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/src/maggotuba/models/train_model.py b/src/maggotuba/models/train_model.py
index 34c87f6..c121ddc 100644
--- a/src/maggotuba/models/train_model.py
+++ b/src/maggotuba/models/train_model.py
@@ -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():
--
GitLab