Merge branch 'js/feature/easy_add_model' into js/feature/add_taskmode…

…l_and_head_factory
nyu-mll · Feb 25, 2021 · f090f1f · f090f1f
2 parents 58b5b18 + b2cfb2a
commit f090f1f
Show file tree

Hide file tree

Showing 39 changed files with 5,279 additions and 3,622 deletions.
diff --git a/examples/notebooks/jiant_Basic_Example.ipynb b/examples/notebooks/jiant_Basic_Example.ipynb
@@ -85,6 +85,7 @@
    "outputs": [],
    "source": [
     "%%capture\n",
+    "%cd /content\n",
     "# Download MRPC data\n",
     "!PYTHONPATH=/content/jiant python jiant/jiant/scripts/download_data/runscript.py \\\n",
     "    download \\\n",
@@ -309,8 +310,8 @@
     "    jiant_task_container_config_path=\"./run_configs/mrpc_run_config.json\",\n",
     "    output_dir=\"./runs/mrpc\",\n",
     "    hf_pretrained_model_name_or_path=\"roberta-base\",\n",
-    "    model_path=\"./models/roberta-base/model/roberta-base.p\",\n",
-    "    model_config_path=\"./models/roberta-base/model/roberta-base.json\",\n",
+    "    model_path=\"./models/roberta-base/model/roberta.p\",\n",
+    "    model_config_path=\"./models/roberta-base/model/roberta.json\",\n",
     "    learning_rate=1e-5,\n",
     "    eval_every_steps=500,\n",
     "    do_train=True,\n",