yujunjun/haystack
1
0
Fork 0
mirror of https://github.com/deepset-ai/haystack.git synced 2025-10-30 17:29:29 +00:00
Code Issues Packages Projects Releases Wiki Activity

Move save_dir from FARMReader() to reader.train()

Browse Source
This commit is contained in:
Branden Chan 2020-05-26 12:14:35 +02:00 committed by GitHub
parent cbe62044b1
commit 5c68a5d755
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
tutorials/Tutorial2_Finetune_a_model_on_your_data.ipynb
View File

@ -90,10 +90,10 @@
}
],
"source": [
"reader = FARMReader(model_name_or_path=\"distilbert-base-uncased-distilled-squad\", use_gpu=False, save_dir=\"my_model\")\n",
"reader = FARMReader(model_name_or_path=\"distilbert-base-uncased-distilled-squad\", use_gpu=False)\n",
"train_data = \"data/squad20\"\n",
"# train_data = \"PATH/TO_YOUR/TRAIN_DATA\" \n",
"reader.train(data_dir=train_data, train_filename=\"dev-v2.0.json\", use_gpu=False, n_epochs=1)"
"reader.train(data_dir=train_data, train_filename=\"dev-v2.0.json\", use_gpu=False, n_epochs=1, save_dir=\"my_model\")"
]
},
{
@ -149,4 +149,4 @@
},
"nbformat": 4,
"nbformat_minor": 2
}
}