From 9ddc0df6bbe176d0d3d7422d5a713a8b73a72a66 Mon Sep 17 00:00:00 2001 From: Sai Pavan Mandava Date: Fri, 10 Jul 2020 22:41:23 +0200 Subject: [PATCH] Config file changes for IMS Machines --- .allennlp_plugins | 1 + configs/basic_model.json | 8 ++++---- 2 files changed, 5 insertions(+), 4 deletions(-) diff --git a/.allennlp_plugins b/.allennlp_plugins index 5ac6fd7..2362f17 100644 --- a/.allennlp_plugins +++ b/.allennlp_plugins @@ -1 +1,2 @@ classifier +utils diff --git a/configs/basic_model.json b/configs/basic_model.json index 369ef62..03ab56c 100644 --- a/configs/basic_model.json +++ b/configs/basic_model.json @@ -10,7 +10,7 @@ "text_field_embedder": { "token_embedders": { "tokens": { - "pretrained_file": "https://s3-us-west-2.amazonaws.com/allennlp/datasets/glove/glove.6B.100d.txt.gz", + "pretrained_file": "/mount/arbeitsdaten/studenten1/team-lab-nlp/mandavsi_rileyic/glove.6B.100d.txt.gz", "type": "embedding", "embedding_dim": 100, "trainable": false @@ -26,7 +26,7 @@ }, "elmo": { "options_file": "https://s3-us-west-2.amazonaws.com/allennlp/models/elmo/2x4096_512_2048cnn_2xhighway_5.5B/elmo_2x4096_512_2048cnn_2xhighway_5.5B_options.json", - "weight_file": "https://s3-us-west-2.amazonaws.com/allennlp/models/elmo/2x4096_512_2048cnn_2xhighway_5.5B/elmo_2x4096_512_2048cnn_2xhighway_5.5B_weights.hdf5", + "weight_file": "/mount/arbeitsdaten/studenten1/team-lab-nlp/mandavsi_rileyic/elmo_2x4096_512_2048cnn_2xhighway_5.5B_weights.hdf5", "do_layer_norm": true, "dropout": 0.5, "num_output_representations": 1 @@ -51,6 +51,6 @@ "lr": 0.001 }, "num_epochs": 2, - "cuda_device": -1 + "cuda_device": 1 } -} \ No newline at end of file +}