From c65c91fc272b4f7c73870ff9f1aaadc248c929b6 Mon Sep 17 00:00:00 2001
From: andrea <andrea.pdr@hotmail.it>
Date: Wed, 3 Feb 2021 12:30:08 +0100
Subject: [PATCH] setting up zero-shot experiments (implemented for Recurrent
 and Bert but not tested for Bert)

---
 main.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/main.py b/main.py
index 5de4b8d..a521fef 100644
--- a/main.py
+++ b/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