diff --git a/cvae.py b/cvae.py
index 7446e9e..f32bf97 100644
--- a/cvae.py
+++ b/cvae.py
@@ -584,6 +584,9 @@ if __name__ == '__main__':
             train_accumulator.reset()
             valid_accumulator.reset()
 
+        if step % 5000 == 0:
+            manager.save(checkpoint_number=step)
+            
         print(step)
 
         step += 1