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()