From bf7a518a0e1f8911ae7851e42f6ce67dfb8e98a5 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Fran=C3=A7ois=20Laurent?= <francois.laurent@posteo.net> Date: Fri, 15 Sep 2023 17:13:20 +0200 Subject: [PATCH] debug messages --- src/maggotuba/models/trainers.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/src/maggotuba/models/trainers.py b/src/maggotuba/models/trainers.py index bc1b1db..e3065cb 100644 --- a/src/maggotuba/models/trainers.py +++ b/src/maggotuba/models/trainers.py @@ -386,17 +386,20 @@ def import_pretrained_model(backend, pretrained_model_instance): for file in pretrained_autoencoder_dir.iterdir(): if not file.is_file(): continue + logging.debug(f"copying file: {file.source}") dst = backend.model_dir() / file.name if file.name.endswith("config.json"): with open(file) as f: config = json.load(f) dir = backend.model_dir().relative_to(backend.project_dir) config["log_dir"] = str(dir) + logging.debug(f"log_dir: \"{config['log_dir']}\"") with open(dst, "w") as f: json.dump(config, f, indent=2) assert config_file is None config_file = dst else: + assert file.name != 'trained_classifier.pt' with open(file, "rb") as i: with open(dst, "wb") as o: o.write(i.read()) -- GitLab