minor fixes

segment-any-text · Apr 1, 2024 · cdcaeba · cdcaeba
1 parent 2ad7843
commit cdcaeba
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/wtpsplit/evaluation/intrinsic_list.py b/wtpsplit/evaluation/intrinsic_list.py
@@ -129,7 +129,7 @@ def load_or_compute_logits(args, model, eval_data, valid_data=None, save_str: st
             for dataset_name, dataset in eval_data[lang_code]["sentence"].items():
                 # train on all mldb, eval on mldbW 
                 if "mldbW" in args.eval_data_path and (
-                    "mldbW" not in args.model_path or "mldbW" not in args.adapter_path
+                    "mldbW" not in args.model_path and "mldbW" not in args.adapter_path
                 ):
                     dataset_load_name = "unk"
                 else:

diff --git a/wtpsplit/train/train_adapter.py b/wtpsplit/train/train_adapter.py
@@ -603,7 +603,9 @@ def compute_metrics(trainer):
         else:
             eval_function = "intrinsic"
         if args.do_lowercase and args.do_remove_punct:
-            suffix = "--do_lowercase --do_remove_punct"  
+            suffix = "--do_lowercase --do_remove_punct" 
+        else:
+            suffix = "" 
         if "adapter" in training_args.output_dir:
             model_info = f"--model_path {args.model_name_or_path} --adapter_path {training_args.output_dir}"
         else: