setting up zero-shot experiments (implemented for Recurrent and Bert but not tested for Bert)
This commit is contained in:
parent
ab3bacb29c
commit
c65c91fc27
2
main.py
2
main.py
|
|
@ -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
|
||||
|
|
|
|||
Loading…
Reference in New Issue