tomaarsen
/

span-marker-bert-small-orgs

Token Classification

named-entity-recognition

generated_from_span_marker_trainer

Model card Files Files and versions Metrics Training metrics Community

tomaarsen HF staff commited on Nov 22, 2023

Commit

0c7c789

•

1 Parent(s): 205bedc

Upload train.py with huggingface_hub

Files changed (1) hide show

train.py +1 -1

train.py CHANGED Viewed

@@ -47,7 +47,7 @@ def main() -> None:
         output_dir=output_dir,
         run_name=model_id,
         # Training Hyperparameters:
-        learning_rate=5e-5,
         per_device_train_batch_size=128,
         per_device_eval_batch_size=128,
         num_train_epochs=3,

         output_dir=output_dir,
         run_name=model_id,
         # Training Hyperparameters:
+        learning_rate=1e-4,
         per_device_train_batch_size=128,
         per_device_eval_batch_size=128,
         num_train_epochs=3,