From b7c83f01a1b00e1a8cb0c68f3d26e6349df43eb0 Mon Sep 17 00:00:00 2001
From: Laurent COLBOIS <laurent.colbois@idiap.ch>
Date: Fri, 18 Jun 2021 13:57:55 +0200
Subject: [PATCH] Fix path in downloaded checkpoints
---
bob/paper/ijcb2021_synthetic_dataset/utils/download.py | 4 ++++
1 file changed, 4 insertions(+)
diff --git a/bob/paper/ijcb2021_synthetic_dataset/utils/download.py b/bob/paper/ijcb2021_synthetic_dataset/utils/download.py
index b84131e..e14c463 100644
--- a/bob/paper/ijcb2021_synthetic_dataset/utils/download.py
+++ b/bob/paper/ijcb2021_synthetic_dataset/utils/download.py
@@ -119,6 +119,10 @@ def download_fr_models(destination_path=FR_MODELS_PATH, **kwargs):
bob.io.base.create_directories_safe(model_path)
bob.extension.download.download_and_unzip([m.url], output_path)
+ checkpoint_path = os.path.join(destination_path, m.group, m.model_name, 'checkpoint')
+ with open(checkpoint_path, 'w') as f:
+ f.write('model_checkpoint_path: "model.ckpt-444274"')
+
# Setting the path
rc[m.config_name] = model_path
bob.extension.rc_config._saverc(rc)
--
GitLab