Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -406,6 +406,7 @@ class TrinerModelVITS:
|
|
406 |
self.load_dataset(train_dataset_dirs,eval_dataset_dir,full_generation_dir)
|
407 |
self.epoch_count=0
|
408 |
self.global_step=0
|
|
|
409 |
|
410 |
|
411 |
|
@@ -415,7 +416,7 @@ class TrinerModelVITS:
|
|
415 |
self.stute_mode=False
|
416 |
print('end training_args')
|
417 |
|
418 |
-
|
419 |
def init_training(self):
|
420 |
|
421 |
self.load_model()
|
|
|
406 |
self.load_dataset(train_dataset_dirs,eval_dataset_dir,full_generation_dir)
|
407 |
self.epoch_count=0
|
408 |
self.global_step=0
|
409 |
+
self.len_dataset=len(self.DataSets['train'])
|
410 |
|
411 |
|
412 |
|
|
|
416 |
self.stute_mode=False
|
417 |
print('end training_args')
|
418 |
|
419 |
+
|
420 |
def init_training(self):
|
421 |
|
422 |
self.load_model()
|