Skip to content

Commit 9f7043f

Browse files
authored
Update peft_whisper.py
1 parent 7e10178 commit 9f7043f

File tree

1 file changed

+16
-16
lines changed

1 file changed

+16
-16
lines changed

examples/peft_whisper.py

Lines changed: 16 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -2,26 +2,26 @@
22

33

44
# Create a dataset instance for the audio files
5-
dataset = Dataset(dummy_data = True, type_ = "full")
5+
dataset = Dataset(dummy_data=True, type_="full")
66

7-
#Model
8-
model = Model("openai/whisper-large-v2", use_peft = True)
7+
# Model
8+
model = Model("openai/whisper-large-v2", use_peft=True)
99

1010
training_args_dict = {
11-
per_device_train_batch_size=16,
12-
gradient_accumulation_steps=1,
13-
learning_rate=1e-3,
14-
warmup_steps=500,
15-
num_train_epochs=1,
16-
evaluation_strategy="steps",
17-
fp16=True,
18-
per_device_eval_batch_size=8,
19-
generation_max_length=225,
20-
logging_steps=100,
21-
max_steps=4000
22-
}
11+
"per_device_train_batch_size": 16,
12+
"gradient_accumulation_steps": 1,
13+
"learning_rate": 1e-3,
14+
"warmup_steps": 500,
15+
"num_train_epochs": 1,
16+
"evaluation_strategy": "steps",
17+
"fp16": True,
18+
"per_device_eval_batch_size": 8,
19+
"generation_max_length": 225,
20+
"logging_steps": 100,
21+
"max_steps": 4000,
22+
}
2323

2424
tuner = Tuner(model, dataset, training_args_dict)
2525
trained_model = tuner.fit()
2626

27-
print(tuner.wer_eval('test'))
27+
print(tuner.wer_eval("test"))

0 commit comments

Comments
 (0)