andrea.pedrotti
/
gFun
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Set arguments in order to reproduce 'master' performances with Neural setting

This commit is contained in:
andrea 2021-01-29 18:25:01 +01:00
parent a301afde97
commit bca0b9ab7c
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
main.py
View File

@ -41,7 +41,7 @@ def main(args):
embedder_list.append(wceEmbedder)
if args.gru_embedder:
rnnEmbedder = RecurrentGen(multilingualIndex, pretrained_embeddings=lMuse, wce=args.gru_wce,
rnnEmbedder = RecurrentGen(multilingualIndex, pretrained_embeddings=lMuse, wce=args.rnn_wce,
batch_size=args.batch_rnn, nepochs=args.nepochs_rnn, patience=args.patience_rnn,
gpus=args.gpus, n_jobs=args.n_jobs)
embedder_list.append(rnnEmbedder)
@ -132,7 +132,7 @@ if __name__ == '__main__':
default=False)
parser.add_argument('-g', '--gru_embedder', dest='gru_embedder', action='store_true',
help='deploy a GRU in order to compute document embeddings',
help='deploy a GRU in order to compute document embeddings (a.k.a., RecurrentGen)',
default=False)
parser.add_argument('-c', '--c_optimize', dest='optimc', action='store_true',
@ -171,8 +171,8 @@ if __name__ == '__main__':
help='Path to the MUSE polylingual word embeddings (default embeddings/)',
default='embeddings/')
parser.add_argument('--gru_wce', dest='gru_wce', action='store_true',
help='Deploy WCE embedding as embedding layer of the GRU View Generator',
parser.add_argument('--rnn_wce', dest='rnn_wce', action='store_true',
help='Deploy WCE embedding as embedding layer of the RecurrentGen',
default=False)
parser.add_argument('--rnn_dir', dest='rnn_dir', type=str, metavar='',