模型优先保存checkpoint_save路径

Qihoo360 · Nov 19, 2021 · 2477156 · 2477156
1 parent b670b1e
commit 2477156
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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()