testing dataset ok; trainer.py take effects

2025-07-11 11:08:51 +08:00
parent 185959cf2a
commit cfa52e5f2e
11 changed files with 167 additions and 97 deletions
--- a/trainer/trainer.py
+++ b/trainer/trainer.py
@@ -26,5 +26,6 @@ def get_config(file_path):

 if __name__ == "__main__":
    # Load configuration
+    # opt = get_config("config_files/4digit_config.yaml")
    opt = get_config("config_files/en_filtered_config.yaml")
    train(opt, amp=False)