From 5bd037e90ae94823641fd68086a3f3157c6407de Mon Sep 17 00:00:00 2001
From: "michael.divia" <michael.divia@etu.hesge.ch>
Date: Wed, 9 Apr 2025 18:24:17 +0200
Subject: [PATCH] Avoid Overfit ResNet50
---
python/pokedex_ResNet50.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/python/pokedex_ResNet50.py b/python/pokedex_ResNet50.py
index 971ac69..2c8d413 100644
--- a/python/pokedex_ResNet50.py
+++ b/python/pokedex_ResNet50.py
@@ -110,7 +110,7 @@ callbacks = [
model.fit(
train_ds,
validation_data=val_ds,
- epochs=20,
+ epochs=5,
callbacks=callbacks,
class_weight=class_weight_dict
)
--
GitLab