snorbyte
/

snorTTS-Indic-v0

text-generation

speech-synthesis

text-generation-inference

Model card Files Files and versions

aarvis commited on Jul 12

Commit

5d4f59d

·

1 Parent(s): 6efa365

update-11

Files changed (1) hide show

train_orpheus.py +2 -2

train_orpheus.py CHANGED Viewed

@@ -38,7 +38,7 @@ if STAGE == 1:
     TRAIN_CSV_PATH = "data/data_stage_1.csv"
     VALID_CSV_PATH = "data/data_eval.csv"
     LR = 2e-4
-    EPOCHS = 1
     MODEL_NAME = f"snorTTS-indicv0-stage-{STAGE}"
 else:
     BASE_MODEL = f"{HUGGINGFACE_USERNAME}/snorTTS-indicv0-stage-1"
@@ -56,7 +56,7 @@ else:
     LR = 2e-4
     EPOCHS = 2
     MODEL_NAME = f"snorTTS-indicv0-stage-{STAGE}"
 TRAIN_NUM_SAMPLES = None
 EVAL_NUM_SAMPLES = 250
 MAX_SEQ_LENGTH = 2048

     TRAIN_CSV_PATH = "data/data_stage_1.csv"
     VALID_CSV_PATH = "data/data_eval.csv"
     LR = 2e-4
+    EPOCHS = 2
     MODEL_NAME = f"snorTTS-indicv0-stage-{STAGE}"
 else:
     BASE_MODEL = f"{HUGGINGFACE_USERNAME}/snorTTS-indicv0-stage-1"
     LR = 2e-4
     EPOCHS = 2
     MODEL_NAME = f"snorTTS-indicv0-stage-{STAGE}"
 TRAIN_NUM_SAMPLES = None
 EVAL_NUM_SAMPLES = 250
 MAX_SEQ_LENGTH = 2048