Update phishing_email_detection_gpt2.py

david-thrower · Nov 17, 2023 · 861e057 · 861e057
1 parent 91412db
commit 861e057
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/phishing_email_detection_gpt2.py b/phishing_email_detection_gpt2.py
@@ -201,7 +201,8 @@ def from_config(cls, config):
     batch_size=batch_size,
     meta_trial_number=meta_trial_number,
     base_models=[base_model],
-    train_data_dtype=tf.string)
+    train_data_dtype=tf.string,
+    jit_compile=False)
 
 result = cerebros_automl.run_random_search()