diff --git a/src/maggotuba/models/trainers.py b/src/maggotuba/models/trainers.py index bc1b1db75aca963396dc48a8e1fe28647f734f02..e3065cb8588f4fbd0a821cb8b42fcf30d3af09c0 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())