diff --git a/pysegcnn/core/trainer.py b/pysegcnn/core/trainer.py index 12cdafbc822843973eeae49db4d93fce1e6572ed..d361ed1c6a5f9a83560a8a0804382f887ea8be1c 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