From 4b7df39facd3047ebc947ac27625a8e3852eb57d Mon Sep 17 00:00:00 2001
From: Starbeamrainbowlabs <sbrl@starbeamrainbowlabs.com>
Date: Mon, 9 Jan 2023 18:25:16 +0000
Subject: [PATCH] bugfix

---
 aimodel/src/deeplabv3_plus_test_rainfall.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/aimodel/src/deeplabv3_plus_test_rainfall.py b/aimodel/src/deeplabv3_plus_test_rainfall.py
index cbe5f83..1af1521 100755
--- a/aimodel/src/deeplabv3_plus_test_rainfall.py
+++ b/aimodel/src/deeplabv3_plus_test_rainfall.py
@@ -153,7 +153,7 @@ history = model.fit(dataset_train,
 			separator="\t"
 		),
 		CallbackCustomModelCheckpoint(
-			model_to_checkpoint=model_predict,
+			model_to_checkpoint=model,
 			filepath=os.path.join(
 				DIR_OUTPUT,
 				"checkpoints"