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 @@ ...@@ -260,7 +260,7 @@
], ],
"source": [ "source": [
"training_args = TrainingArguments(\n", "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", " num_train_epochs=30, # total # of training epochs\n",
" per_device_train_batch_size=1, # batch size per device during training\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", " 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