From 1b58fed14d1ab31ee7057067e93c96bd7df143e3 Mon Sep 17 00:00:00 2001 From: andreapdr Date: Sun, 29 Oct 2023 18:18:01 +0100 Subject: [PATCH] removed unused --- infer.py | 2 -- 1 file changed, 2 deletions(-) diff --git a/infer.py b/infer.py index b15f94d..39bffce 100644 --- a/infer.py +++ b/infer.py @@ -25,7 +25,6 @@ def main(args): textual_transformer=True, load_trained=args.trained_gfun, load_meta=True, - trained_text_trf=args.trained_transformer, ) predictions = gfun.transform(lX) @@ -42,7 +41,6 @@ if __name__ == "__main__": parser.add_argument("--nlabels", type=int, default=28) parser.add_argument("--muse_dir", type=str, default="~/resources/muse_embeddings", help="Path to muse embeddings") parser.add_argument("--trained_gfun", type=str, default="rai_pmt_mean_231029", help="name of the trained gfun instance") - parser.add_argument("--trained_transformer", default="hf_models/mbert-fewshot-rai-full/checkpoint-5150-small", help="path to fine-tuned transformer") args = parser.parse_args() main(args)