refactor: Use AdamW optimizer and increase early stopping patience

2025-10-17 10:31:12 +08:00
parent cb7575a229
commit 02d84a7eca
2 changed files with 7 additions and 3 deletions
--- a/requirements.txt
+++ b/requirements.txt
@@ -0,0 +1,4 @@
+torch
+numpy
+tqdm
+matplotlib