Skip to content

Commit 56ceae5

Browse files
authored
Updating notebooks, removing model_type (#1270)
* Updating notebooks, fixing model_type * add gpu to test
1 parent 723786a commit 56ceae5

File tree

9 files changed

+3449
-3444
lines changed

9 files changed

+3449
-3444
lines changed

examples/notebooks/jiant_Basic_Example.ipynb

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -85,6 +85,7 @@
8585
"outputs": [],
8686
"source": [
8787
"%%capture\n",
88+
"%cd /content\n",
8889
"# Download MRPC data\n",
8990
"!PYTHONPATH=/content/jiant python jiant/jiant/scripts/download_data/runscript.py \\\n",
9091
" download \\\n",
@@ -309,8 +310,8 @@
309310
" jiant_task_container_config_path=\"./run_configs/mrpc_run_config.json\",\n",
310311
" output_dir=\"./runs/mrpc\",\n",
311312
" hf_pretrained_model_name_or_path=\"roberta-base\",\n",
312-
" model_path=\"./models/roberta-base/model/roberta-base.p\",\n",
313-
" model_config_path=\"./models/roberta-base/model/roberta-base.json\",\n",
313+
" model_path=\"./models/roberta-base/model/roberta.p\",\n",
314+
" model_config_path=\"./models/roberta-base/model/roberta.json\",\n",
314315
" learning_rate=1e-5,\n",
315316
" eval_every_steps=500,\n",
316317
" do_train=True,\n",

0 commit comments

Comments
 (0)