|
|
@@ -160,14 +160,13 @@ class _TrainerMixin(BaseMixin):
|
|
|
return Path(self.trainer.out, name)
|
|
|
|
|
|
def dump(self, suffix):
|
|
|
- self._check_attr("config")
|
|
|
- if self.only_eval or self.no_snapshot:
|
|
|
+ self._check_attr("only_eval")
|
|
|
+ if self.only_eval:
|
|
|
return
|
|
|
|
|
|
clf_file = self._trainer_output(f"clf_{suffix}.npz")
|
|
|
logging.info(f"Storing classifier weights to {clf_file}")
|
|
|
save_npz(clf_file, self.clf)
|
|
|
- # save_npz(self._trainer_output(f"model_{suffix}.npz"), self.model)
|
|
|
|
|
|
def save_meta_info(self, meta_folder: str = "meta"):
|
|
|
self._check_attr("config")
|
