Christopher H. commited on
Update Readme
Browse files
README.md
CHANGED
|
@@ -109,9 +109,10 @@ model.freeze_decoder()
|
|
| 109 |
|
| 110 |
# Define training arguments
|
| 111 |
training_args = TrainingArguments(
|
| 112 |
-
|
|
|
|
| 113 |
|
| 114 |
-
num_train_epochs=
|
| 115 |
per_device_train_batch_size=256,
|
| 116 |
per_device_eval_batch_size=386,
|
| 117 |
|
|
@@ -120,7 +121,6 @@ training_args = TrainingArguments(
|
|
| 120 |
weight_decay=0.01,
|
| 121 |
|
| 122 |
# Logging settings
|
| 123 |
-
logging_dir="./logs",
|
| 124 |
logging_steps=500,
|
| 125 |
logging_strategy="steps",
|
| 126 |
|
|
|
|
| 109 |
|
| 110 |
# Define training arguments
|
| 111 |
training_args = TrainingArguments(
|
| 112 |
+
hub_model_id="mrprimenotes/sign-whisper-german_trained",
|
| 113 |
+
push_to_hub=True,
|
| 114 |
|
| 115 |
+
num_train_epochs=2,
|
| 116 |
per_device_train_batch_size=256,
|
| 117 |
per_device_eval_batch_size=386,
|
| 118 |
|
|
|
|
| 121 |
weight_decay=0.01,
|
| 122 |
|
| 123 |
# Logging settings
|
|
|
|
| 124 |
logging_steps=500,
|
| 125 |
logging_strategy="steps",
|
| 126 |
|