mrprimenotes commited on
Commit
af47a58
·
verified ·
1 Parent(s): 6983214

Update Readme

Browse files
Files changed (1) hide show
  1. README.md +3 -3
README.md CHANGED
@@ -109,9 +109,10 @@ model.freeze_decoder()
109
 
110
  # Define training arguments
111
  training_args = TrainingArguments(
112
- output_dir="./sign-whisper-german",
 
113
 
114
- num_train_epochs=3,
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