From fdbfcf3b67ef0a82dcd10cba3b55def4784b7822 Mon Sep 17 00:00:00 2001
From: "Daniel.Frisinghelli" <daniel.frisinghelli@eurac.edu>
Date: Wed, 3 Mar 2021 16:34:03 +0100
Subject: [PATCH] Fixed model name for supervised transfer learning.
---
pysegcnn/core/trainer.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/pysegcnn/core/trainer.py b/pysegcnn/core/trainer.py
index 12cdafb..d361ed1 100644
--- a/pysegcnn/core/trainer.py
+++ b/pysegcnn/core/trainer.py
@@ -816,7 +816,7 @@ class StateConfig(BaseConfig):
# state file for models fine-tuned to target domain
# PretrainedModel_DatasetConfig.pt
state = '_'.join([
- tc.pretrained_model, 'sda_{}'.format(src_ds_state),
+ tc.pretrained_path.stem, 'sda_{}'.format(src_ds_state),
src_ds_ext])
else:
# state file for models trained via unsupervised domain
--
GitLab