andrea.pedrotti
/
gfun_multimodal
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

removed unused

This commit is contained in:
Andrea Pedrotti 2023-10-29 18:18:01 +01:00
parent 41ba20ad5c
commit 1b58fed14d
1 changed files with 0 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
infer.py
View File

@ -25,7 +25,6 @@ def main(args):
textual_transformer=True, textual_transformer=True,
load_trained=args.trained_gfun, load_trained=args.trained_gfun,
load_meta=True, load_meta=True,
trained_text_trf=args.trained_transformer,
) )
predictions = gfun.transform(lX) predictions = gfun.transform(lX)
@ -42,7 +41,6 @@ if __name__ == "__main__":
parser.add_argument("--nlabels", type=int, default=28) 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("--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_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() args = parser.parse_args()
main(args) main(args)