yujunjun/haystack
1
0
Fork 0
mirror of https://github.com/deepset-ai/haystack.git synced 2025-11-01 02:09:39 +00:00
Code Issues Packages Projects Releases Wiki Activity

Remove use_gpu param

Browse Source
This commit is contained in:
Branden Chan 2020-03-24 17:47:00 +01:00
parent b5c58b32f0
commit 420e11695b
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tutorials/Tutorial2_Finetune_a_model_on_your_data.py
View File

@ -13,7 +13,7 @@ reader = FARMReader(model_name_or_path="distilbert-base-uncased-distilled-squad"
# and fine-tune it on your own custom dataset (should be in SQuAD like format)
train_data = "PATH/TO_YOUR/TRAIN_DATA"
reader.train(data_dir=train_data, train_filename="train.json", use_gpu=False, n_epochs=1)
reader.train(data_dir=train_data, train_filename="train.json", n_epochs=1)
#### Use it (same as in Tutorial 1) #############