Skip to content

Commit 4440719

Browse files
Update phishing_email_detection_gpt2.py
Tiny scale test to accelerate test completion and validate that updates in #224 work ....
1 parent 495c0ae commit 4440719

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

phishing_email_detection_gpt2.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -75,7 +75,7 @@
7575
minimum_neurons_per_unit = 1
7676
maximum_neurons_per_unit = 2
7777

78-
moities_to_try = 5
78+
moities_to_try = 3
7979
tries_per_moity = 1
8080

8181
####### DO NOT FORGET TO MERGE IN THE WORK THAT ADDED GRADIENT
@@ -231,7 +231,7 @@ def package_non_instruct_text(text: str, desired_samples: int, max_length_tokens
231231
return samples
232232

233233
# Separate into samples
234-
non_instruct_samples = package_non_instruct_text(text=bible, desired_samples=30, max_length_tokens=int(np.ceil(MAX_SEQ_LENGTH * .8))) ##
234+
non_instruct_samples = package_non_instruct_text(text=bible, desired_samples=10, max_length_tokens=int(np.ceil(MAX_SEQ_LENGTH * .8))) ##
235235

236236
del(bible)
237237
collect()

0 commit comments

Comments
 (0)