Commit 972bc6ef authored by Bandara I.M.R.H IT18013474's avatar Bandara I.M.R.H IT18013474 💬

Update Public_Violation_Detection.ipynb

parent 003805d7
......@@ -260,7 +260,7 @@
],
"source": [
"training_args = TrainingArguments(\n",
" output_dir='Public Violation Detection', # output directory\n",
" output_dir='Public Violation Detection', # output directory\n",
" num_train_epochs=30, # total # of training epochs\n",
" per_device_train_batch_size=1, # batch size per device during training\n",
" per_device_eval_batch_size=1, # batch size for evaluation\n",
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment