Disfluency-large

Running

DD0101 commited on May 3, 2023

Commit

034bc4e

1 Parent(s): c3dbea2

changing data_dir

Files changed (1) hide show

Customized_IDSF/load_model.py CHANGED Viewed

@@ -22,7 +22,7 @@ def get_args(pred_config):
     args = torch.load(os.path.join(pred_config.model_dir, "training_args.bin"))
     args.model_dir = pred_config.model_dir
-    args.data_dir = './Customized_IDSF/PhoATIS'
     return args

     args = torch.load(os.path.join(pred_config.model_dir, "training_args.bin"))
     args.model_dir = pred_config.model_dir
+    args.data_dir = '/home/user/app/Customized_IDSF/PhoATIS'
     return args