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