File tree Expand file tree Collapse file tree 1 file changed +16
-16
lines changed Expand file tree Collapse file tree 1 file changed +16
-16
lines changed Original file line number Diff line number Diff line change 2
2
3
3
4
4
# Create a dataset instance for the audio files
5
- dataset = Dataset (dummy_data = True , type_ = "full" )
5
+ dataset = Dataset (dummy_data = True , type_ = "full" )
6
6
7
- #Model
8
- model = Model ("openai/whisper-large-v2" , use_peft = True )
7
+ # Model
8
+ model = Model ("openai/whisper-large-v2" , use_peft = True )
9
9
10
10
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
+ }
23
23
24
24
tuner = Tuner (model , dataset , training_args_dict )
25
25
trained_model = tuner .fit ()
26
26
27
- print (tuner .wer_eval (' test' ))
27
+ print (tuner .wer_eval (" test" ))
You can’t perform that action at this time.
0 commit comments