From 2d2dcb26ca9811b074e8b9013dafec3f9b455367 Mon Sep 17 00:00:00 2001
From: brunomachado37 <brunomachadocarneiro37@gmail.com>
Date: Wed, 19 Mar 2025 07:00:01 +0100
Subject: [PATCH] Add instruction
---
BE2_GAN_and_Diffusion.ipynb | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/BE2_GAN_and_Diffusion.ipynb b/BE2_GAN_and_Diffusion.ipynb
index d018170..f72c2bd 100644
--- a/BE2_GAN_and_Diffusion.ipynb
+++ b/BE2_GAN_and_Diffusion.ipynb
@@ -973,7 +973,7 @@
" \n",
" if (epoch + 1) % 100 == 0:\n",
" print('Saving model...')\n",
- " save_model(epoch, loss_D, loss_G)\n"
+ " save_model(epoch+1, loss_D, loss_G)\n"
]
},
{
--
GitLab