From 4c6497f726fd0af540050507336bffbf46c612d7 Mon Sep 17 00:00:00 2001 From: Hunor Laczko Date: Tue, 28 Feb 2023 20:04:30 +0000 Subject: [PATCH] fix --- cvae_pyramid.py | 3 +++ 1 file changed, 3 insertions(+) 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