From 07fef2b8fa39430710bc21d6c5da67aafffe0cfd Mon Sep 17 00:00:00 2001 From: GilesStrong Date: Thu, 12 Nov 2020 18:42:59 +0100 Subject: [PATCH] Fix cb_savepath creation --- CHANGES.md | 2 ++ lumin/nn/models/model.py | 1 + 2 files changed, 3 insertions(+) diff --git a/CHANGES.md b/CHANGES.md index 1db8dd8..157a4fe 100644 --- a/CHANGES.md +++ b/CHANGES.md @@ -10,6 +10,8 @@ ## Fixes +- `Model` now creats `cb_savepath` is it didn't already exist + ## Changes ## Depreciations diff --git a/lumin/nn/models/model.py b/lumin/nn/models/model.py index e92f735..3204fe1 100644 --- a/lumin/nn/models/model.py +++ b/lumin/nn/models/model.py @@ -179,6 +179,7 @@ def fit(self, n_epochs:int, fy:FoldYielder, bs:int, bulk_move:bool=True, train_o self.fit_params = FitParams(cbs=cbs, cyclic_cbs=cyclic_cbs, loss_cbs=loss_cbs, stop=False, n_epochs=n_epochs, fy=fy, val_idx=val_idx, bs=bs, bulk_move=bulk_move, train_on_weights=train_on_weights, cb_savepath=Path(cb_savepath), loss_func=self.loss, opt=self.opt) + self.fit_params.cb_savepath.mkdir(parents=True, exist_ok=True) if inspect.isclass(self.fit_params.loss_func) or isinstance(self.fit_params.loss_func, partial): self.fit_params.loss_func = self.fit_params.loss_func() self.fit_params.partial_by = partialler(BatchYielder, objective=self.objective, use_weights=self.fit_params.train_on_weights, bulk_move=self.fit_params.bulk_move, input_mask=self.input_mask)