add data folder to parameters
"outputs": [],
"source": [
"# Parameters for notebook testing.\n",
+ "DATA_FOLDER = './squad'\n",
"BERT_MODEL = 'bert-large-uncased'\n",
"NUM_TRAIN_EPOCHS = 2.0\n",
"NODE_COUNT = 2\n",
"metadata": {},
"outputs": [],
"source": [
- "data_folder = './squad'"
+ "data_folder = DATA_FOLDER"
]
},
{
