setting up zero-shot experiments (implemented for Recurrent and Bert but not tested for Bert)

This commit is contained in:
andrea 2021-02-03 12:30:08 +01:00
parent ab3bacb29c
commit c65c91fc27
1 changed files with 1 additions and 1 deletions

View File

@ -54,8 +54,8 @@ def main(args):
if args.bert_embedder:
bertEmbedder = BertGen(multilingualIndex, batch_size=args.batch_bert, nepochs=args.nepochs_bert,
zero_shot=zero_shot, train_langs=zscl_train_langs, # Todo: testing zero shot
patience=args.patience_bert, gpus=args.gpus, n_jobs=args.n_jobs)
bertEmbedder.transform(lX)
embedder_list.append(bertEmbedder)
# Init DocEmbedderList (i.e., first-tier learners or view generators) and metaclassifier