mrprimenotes
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 |
|