diff --git a/cvae_pyramid.py b/cvae_pyramid.py
index d1287be..2918afd 100644
--- a/cvae_pyramid.py
+++ b/cvae_pyramid.py
@@ -880,6 +880,9 @@ if __name__ == '__main__':
             train_accumulator.reset()
             valid_accumulator.reset()
 
+        if step % 5000 == 0:
+            manager.save(checkpoint_number=step)
+            
         print(step)
 
         step += 1