Update phishing_email_detection_gpt2.py

david-thrower · web-flow · commit a58c60e74cd2 · 2025-09-14T15:08:18.000-04:00
Lower gradient_accumulation_steps to 2.
diff --git a/phishing_email_detection_gpt2.py b/phishing_email_detection_gpt2.py
@@ -65,7 +65,7 @@
 learning_rate = 3 * 10 ** -4 # 3 * 10 ** -3
 epochs = 15  #
 batch_size = 10 # 17
-gradient_accumulation_steps = 3
+gradient_accumulation_steps = 2
 minimum_levels = 2
 maximum_levels = 2 # [3,7]
 
