diff options
Diffstat (limited to 'models')
| -rw-r--r-- | models/model.py | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/models/model.py b/models/model.py index 5ddcf90..c0adc41 100644 --- a/models/model.py +++ b/models/model.py @@ -82,7 +82,9 @@ class Model: @property def _model_sig(self) -> str: - return '_'.join((self._model_name, self.curr_iter, self.total_iter)) + return '_'.join( + (self._model_name, str(self.curr_iter), str(self.total_iter)) + ) @property def _checkpoint_sig(self) -> str: @@ -95,7 +97,7 @@ class Model: @property def _log_sig(self) -> str: - return '_'.join((self._model_name, self.total_iter, self._hp_sig, + return '_'.join((self._model_name, str(self.total_iter), self._hp_sig, self._dataset_sig, str(self.pr), str(self.k))) @property |