Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -575,7 +575,7 @@ ctrain_datasets,eval_dataset,full_generation_dataset=get_data_loader(train_datas
|
|
575 |
|
576 |
|
577 |
|
578 |
-
|
579 |
training_args.d_learning_rate=2e-4
|
580 |
training_args.learning_rate=2e-4
|
581 |
training_args.weight_mel=45
|
@@ -592,7 +592,7 @@ wandb.init(project= 'AZ',config = training_args.to_dict())
|
|
592 |
for i in range(10000):
|
593 |
# model.train(True)
|
594 |
print(f'clcye epochs ={i}')
|
595 |
-
model=VitsModel.from_pretrained(dir_model).to(device)
|
596 |
# model.setMfA(monotonic_align.maximum_path)
|
597 |
#dir_model_save=dir_model+'/vend'
|
598 |
|
|
|
575 |
|
576 |
|
577 |
|
578 |
+
training_args.weight_kl=1
|
579 |
training_args.d_learning_rate=2e-4
|
580 |
training_args.learning_rate=2e-4
|
581 |
training_args.weight_mel=45
|
|
|
592 |
for i in range(10000):
|
593 |
# model.train(True)
|
594 |
print(f'clcye epochs ={i}')
|
595 |
+
model=VitsModel.from_pretrained(dir_model,token=token).to(device)
|
596 |
# model.setMfA(monotonic_align.maximum_path)
|
597 |
#dir_model_save=dir_model+'/vend'
|
598 |
|