opencampus
/

sign-whisper-german

Inference Endpoints

Model card Files Files and versions Community

mrprimenotes commited on Jan 5

Commit

860320d

·

verified ·

1 Parent(s): d9ae039

Update README.md

Files changed (1) hide show

README.md +3 -0

README.md CHANGED Viewed

@@ -106,9 +106,12 @@ val_dataset = YourSignDataset(...)
 # Define training arguments
 training_args = TrainingArguments(
     output_dir="./sign-whisper-german",
     num_train_epochs=3,
     per_device_train_batch_size=1024,
     per_device_eval_batch_size=256,
     warmup_steps=500,
     weight_decay=0.01,

 # Define training arguments
 training_args = TrainingArguments(
     output_dir="./sign-whisper-german",
     num_train_epochs=3,
     per_device_train_batch_size=1024,
     per_device_eval_batch_size=256,
+    learning_rate=1e-5
     warmup_steps=500,
     weight_decay=0.01,