From 2477156eced2a6ae6ca98b763f7ab225b35c7ead Mon Sep 17 00:00:00 2001 From: wangxingda Date: Fri, 19 Nov 2021 11:54:18 +0800 Subject: [PATCH] =?UTF-8?q?=E6=A8=A1=E5=9E=8B=E4=BC=98=E5=85=88=E4=BF=9D?= =?UTF-8?q?=E5=AD=98checkpoint=5Fsave=E8=B7=AF=E5=BE=84?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- tensornet/callbacks/callbacks.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tensornet/callbacks/callbacks.py b/tensornet/callbacks/callbacks.py index be37c98..eaaa32f 100644 --- a/tensornet/callbacks/callbacks.py +++ b/tensornet/callbacks/callbacks.py @@ -63,7 +63,7 @@ def on_train_end(self, logs=None): self.model.show_decay(self.delta_days) if not self.need_save_model: return - self.model.save_weights(self.checkpoint_dir, dt=self.dt, mode=self.save_mode) + self.model.save_weights(self.checkpoint_save, dt=self.dt, mode=self.save_mode) def on_predict_begin(self, logs=None): self.load_model()