Add trained xTTS model

best_model.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:d13484d2e09730daa3dc3ea46b28eee10d4bc45d4dbf42d190865b99e0b81ca1
+size 5640358637

best_model_11584.pth

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:d13484d2e09730daa3dc3ea46b28eee10d4bc45d4dbf42d190865b99e0b81ca1
+size 5640358637

config.json

@@ -0,0 +1,193 @@
+{
+    "output_path": "xTTS/checkpoints/",
+    "logger_uri": null,
+    "run_name": "GPT_XTTS_FT",
+    "project_name": "XTTS_trainer",
+    "run_description": [
+        "\n        GPT XTTS training\n        "
+    ],
+    "print_step": 50,
+    "plot_step": 100,
+    "model_param_stats": false,
+    "wandb_entity": null,
+    "dashboard_logger": "tensorboard",
+    "save_on_interrupt": true,
+    "log_model_step": 100,
+    "save_step": 9999999999,
+    "save_n_checkpoints": 1,
+    "save_checkpoints": false,
+    "save_all_best": false,
+    "save_best_after": 10000,
+    "target_loss": null,
+    "print_eval": false,
+    "test_delay_epochs": 0,
+    "run_eval": true,
+    "run_eval_steps": null,
+    "distributed_backend": "nccl",
+    "distributed_url": "tcp://localhost:54321",
+    "mixed_precision": false,
+    "precision": "fp16",
+    "epochs": 2,
+    "batch_size": 4,
+    "eval_batch_size": 16,
+    "grad_clip": 0.0,
+    "scheduler_after_epoch": true,
+    "lr": 5e-06,
+    "optimizer": "AdamW",
+    "optimizer_params": {
+        "betas": [
+            0.9,
+            0.96
+        ],
+        "eps": 1e-08,
+        "weight_decay": 0.01
+    },
+    "lr_scheduler": "MultiStepLR",
+    "lr_scheduler_params": {
+        "milestones": [
+            900000,
+            2700000,
+            5400000
+        ],
+        "gamma": 0.5,
+        "last_epoch": -1
+    },
+    "use_grad_scaler": false,
+    "allow_tf32": false,
+    "cudnn_enable": true,
+    "cudnn_deterministic": false,
+    "cudnn_benchmark": false,
+    "training_seed": 54321,
+    "model": "xtts",
+    "num_loader_workers": 8,
+    "num_eval_loader_workers": 0,
+    "use_noise_augment": false,
+    "audio": {
+        "sample_rate": 22050,
+        "output_sample_rate": 24000,
+        "dvae_sample_rate": 22050
+    },
+    "use_phonemes": false,
+    "phonemizer": null,
+    "phoneme_language": null,
+    "compute_input_seq_cache": false,
+    "text_cleaner": null,
+    "enable_eos_bos_chars": false,
+    "test_sentences_file": "",
+    "phoneme_cache_path": null,
+    "characters": null,
+    "add_blank": false,
+    "batch_group_size": 0,
+    "loss_masking": null,
+    "min_audio_len": 1,
+    "max_audio_len": Infinity,
+    "min_text_len": 1,
+    "max_text_len": Infinity,
+    "compute_f0": false,
+    "compute_energy": false,
+    "compute_linear_spec": false,
+    "precompute_num_workers": 0,
+    "start_by_longest": false,
+    "shuffle": false,
+    "drop_last": false,
+    "datasets": [
+        {
+            "formatter": "",
+            "dataset_name": "",
+            "path": "",
+            "meta_file_train": "",
+            "ignored_speakers": null,
+            "language": "",
+            "phonemizer": "",
+            "meta_file_val": "",
+            "meta_file_attn_mask": ""
+        }
+    ],
+    "test_sentences": [],
+    "eval_split_max_size": 256,
+    "eval_split_size": 0.01,
+    "use_speaker_weighted_sampler": false,
+    "speaker_weighted_sampler_alpha": 1.0,
+    "use_language_weighted_sampler": false,
+    "language_weighted_sampler_alpha": 1.0,
+    "use_length_weighted_sampler": false,
+    "length_weighted_sampler_alpha": 1.0,
+    "model_args": {
+        "gpt_batch_size": 1,
+        "enable_redaction": false,
+        "kv_cache": true,
+        "gpt_checkpoint": "",
+        "clvp_checkpoint": null,
+        "decoder_checkpoint": null,
+        "num_chars": 255,
+        "tokenizer_file": "xTTS/checkpoints/XTTS_v2.0_original_model_files/vocab.json",
+        "gpt_max_audio_tokens": 605,
+        "gpt_max_text_tokens": 402,
+        "gpt_max_prompt_tokens": 70,
+        "gpt_layers": 30,
+        "gpt_n_model_channels": 1024,
+        "gpt_n_heads": 16,
+        "gpt_number_text_tokens": 8000,
+        "gpt_start_text_token": 261,
+        "gpt_stop_text_token": 0,
+        "gpt_num_audio_tokens": 1026,
+        "gpt_start_audio_token": 1024,
+        "gpt_stop_audio_token": 1025,
+        "gpt_code_stride_len": 1024,
+        "gpt_use_masking_gt_prompt_approach": true,
+        "gpt_use_perceiver_resampler": true,
+        "input_sample_rate": 22050,
+        "output_sample_rate": 24000,
+        "output_hop_length": 256,
+        "decoder_input_dim": 1024,
+        "d_vector_dim": 512,
+        "cond_d_vector_in_each_upsampling_layer": true,
+        "duration_const": 102400,
+        "min_conditioning_length": 11025,
+        "max_conditioning_length": 132300,
+        "gpt_loss_text_ce_weight": 0.01,
+        "gpt_loss_mel_ce_weight": 1.0,
+        "debug_loading_failures": false,
+        "max_wav_length": 320000,
+        "max_text_length": 400,
+        "mel_norm_file": "xTTS/checkpoints/XTTS_v2.0_original_model_files/mel_stats.pth",
+        "dvae_checkpoint": "xTTS/checkpoints/XTTS_v2.0_original_model_files/dvae.pth",
+        "xtts_checkpoint": "xTTS/checkpoints/XTTS_v2.0_original_model_files/model.pth",
+        "vocoder": ""
+    },
+    "model_dir": null,
+    "languages": [
+        "en",
+        "es",
+        "fr",
+        "de",
+        "it",
+        "pt",
+        "pl",
+        "tr",
+        "ru",
+        "nl",
+        "cs",
+        "ar",
+        "zh-cn",
+        "hu",
+        "ko",
+        "ja",
+        "hi",
+        "wo"
+    ],
+    "temperature": 0.75,
+    "length_penalty": 1.0,
+    "repetition_penalty": 5.0,
+    "top_k": 50,
+    "top_p": 0.85,
+    "num_gpt_outputs": 1,
+    "gpt_cond_len": 30,
+    "gpt_cond_chunk_len": 4,
+    "max_ref_len": 30,
+    "sound_norm_refs": false,
+    "optimizer_wd_only_on_weights": true,
+    "weighted_loss_attrs": null,
+    "weighted_loss_multipliers": null,
+    "github_branch": "inside_docker"
+}

events.out.tfevents.1748866975.eafda0218e5b.317.0

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:92bbd9c8257e7635c9e148d4d60f774e3b306139eaa7dbc879c9709a077dd17f
+size 60608

train_gpt_xtts.py

@@ -0,0 +1,232 @@
+import os
+import gc
+
+from trainer import Trainer, TrainerArgs
+
+from TTS.config.shared_configs import BaseDatasetConfig
+from TTS.tts.datasets import load_tts_samples
+from TTS.tts.layers.xtts.trainer.gpt_trainer import GPTArgs, GPTTrainer, GPTTrainerConfig, XttsAudioConfig
+from TTS.utils.manage import ModelManager
+
+from dataclasses import dataclass, field
+from typing import Optional
+from transformers import HfArgumentParser
+
+import argparse
+
+def create_xtts_trainer_parser():
+    parser = argparse.ArgumentParser(description="Arguments for XTTS Trainer")
+
+    parser.add_argument("--output_path", type=str, required=True,
+                        help="Path to pretrained + checkpoint model")
+    parser.add_argument("--metadatas", nargs='+', type=str, required=True,
+                        help="train_csv_path,eval_csv_path,language")
+    parser.add_argument("--num_epochs", type=int, default=1,
+                        help="Number of epochs")
+    parser.add_argument("--batch_size", type=int, default=1,
+                        help="Mini batch size")
+    parser.add_argument("--grad_acumm", type=int, default=1,
+                        help="Grad accumulation steps")
+    parser.add_argument("--max_audio_length", type=int, default=255995,
+                        help="Max audio length")
+    parser.add_argument("--max_text_length", type=int, default=200,
+                        help="Max text length")
+    parser.add_argument("--weight_decay", type=float, default=1e-2,
+                        help="Weight decay")
+    parser.add_argument("--lr", type=float, default=5e-6,
+                        help="Learning rate")
+    parser.add_argument("--save_step", type=int, default=5000,
+                        help="Save step")
+
+    return parser
+
+
+
+def  train_gpt(metadatas, num_epochs, batch_size, grad_acumm, output_path, max_audio_length, max_text_length, lr, weight_decay, save_step):
+    #  Logging parameters
+    RUN_NAME = "GPT_XTTS_FT"
+    PROJECT_NAME = "XTTS_trainer"
+    DASHBOARD_LOGGER = "tensorboard"
+    LOGGER_URI = None
+
+    # Set here the path that the checkpoints will be saved. Default: ./run/training/
+    # OUT_PATH = os.path.join(output_path, "run", "training")
+    OUT_PATH = output_path
+
+    # Training Parameters
+    OPTIMIZER_WD_ONLY_ON_WEIGHTS = True  # for multi-gpu training please make it False
+    START_WITH_EVAL = False  # if True it will star with evaluation
+    BATCH_SIZE = batch_size  # set here the batch size
+    GRAD_ACUMM_STEPS = grad_acumm  # set here the grad accumulation steps
+
+
+    # Define here the dataset that you want to use for the fine-tuning on.
+    DATASETS_CONFIG_LIST = []
+    for metadata in metadatas:
+        train_csv, eval_csv, language = metadata.split(",")
+        print(train_csv, eval_csv, language)
+
+        config_dataset = BaseDatasetConfig(
+            formatter="coqui",
+            dataset_name="ft_dataset",
+            path=os.path.dirname(train_csv),
+            meta_file_train=os.path.basename(train_csv),
+            meta_file_val=os.path.basename(eval_csv),
+            language=language,
+        )
+
+        DATASETS_CONFIG_LIST.append(config_dataset)
+
+    # Define the path where XTTS v2.0.1 files will be downloaded
+    CHECKPOINTS_OUT_PATH = os.path.join(OUT_PATH, "XTTS_v2.0_original_model_files/")
+    os.makedirs(CHECKPOINTS_OUT_PATH, exist_ok=True)
+
+
+    # DVAE files
+    DVAE_CHECKPOINT_LINK = "https://coqui.gateway.scarf.sh/hf-coqui/XTTS-v2/main/dvae.pth"
+    MEL_NORM_LINK = "https://coqui.gateway.scarf.sh/hf-coqui/XTTS-v2/main/mel_stats.pth"
+
+    # Set the path to the downloaded files
+    DVAE_CHECKPOINT = os.path.join(CHECKPOINTS_OUT_PATH, os.path.basename(DVAE_CHECKPOINT_LINK))
+    MEL_NORM_FILE = os.path.join(CHECKPOINTS_OUT_PATH, os.path.basename(MEL_NORM_LINK))
+
+    # download DVAE files if needed
+    if not os.path.isfile(DVAE_CHECKPOINT) or not os.path.isfile(MEL_NORM_FILE):
+        print(" > Downloading DVAE files!")
+        ModelManager._download_model_files([MEL_NORM_LINK, DVAE_CHECKPOINT_LINK], CHECKPOINTS_OUT_PATH, progress_bar=True)
+
+
+    # Download XTTS v2.0 checkpoint if needed
+    TOKENIZER_FILE_LINK = "https://coqui.gateway.scarf.sh/hf-coqui/XTTS-v2/main/vocab.json"
+    XTTS_CHECKPOINT_LINK = "https://coqui.gateway.scarf.sh/hf-coqui/XTTS-v2/main/model.pth"
+    XTTS_CONFIG_LINK = "https://coqui.gateway.scarf.sh/hf-coqui/XTTS-v2/main/config.json"
+
+    # XTTS transfer learning parameters: You we need to provide the paths of XTTS model checkpoint that you want to do the fine tuning.
+    TOKENIZER_FILE = os.path.join(CHECKPOINTS_OUT_PATH, os.path.basename(TOKENIZER_FILE_LINK))  # vocab.json file
+    XTTS_CHECKPOINT = os.path.join(CHECKPOINTS_OUT_PATH, os.path.basename(XTTS_CHECKPOINT_LINK))  # model.pth file
+    XTTS_CONFIG_FILE = os.path.join(CHECKPOINTS_OUT_PATH, os.path.basename(XTTS_CONFIG_LINK))  # config.json file
+
+    # download XTTS v2.0 files if needed
+    if not os.path.isfile(TOKENIZER_FILE):
+        print(" > Downloading XTTS v2.0 tokenizer!")
+        ModelManager._download_model_files(
+            [TOKENIZER_FILE_LINK], CHECKPOINTS_OUT_PATH, progress_bar=True
+        )
+    if not os.path.isfile(XTTS_CHECKPOINT):
+        print(" > Downloading XTTS v2.0 checkpoint!")
+        ModelManager._download_model_files(
+            [XTTS_CHECKPOINT_LINK], CHECKPOINTS_OUT_PATH, progress_bar=True
+        )
+    if not os.path.isfile(XTTS_CONFIG_FILE):
+        print(" > Downloading XTTS v2.0 config!")
+        ModelManager._download_model_files(
+            [XTTS_CONFIG_LINK], CHECKPOINTS_OUT_PATH, progress_bar=True
+        )
+
+    # init args and config
+    model_args = GPTArgs(
+        max_conditioning_length=132300,  # 6 secs
+        min_conditioning_length=11025,  # 0.5 secs
+        debug_loading_failures=False,
+        max_wav_length=max_audio_length,  # ~11.6 seconds
+        max_text_length=max_text_length,
+        mel_norm_file=MEL_NORM_FILE,
+        dvae_checkpoint=DVAE_CHECKPOINT,
+        xtts_checkpoint=XTTS_CHECKPOINT,  # checkpoint path of the model that you want to fine-tune
+        tokenizer_file=TOKENIZER_FILE,
+        gpt_num_audio_tokens=1026,
+        gpt_start_audio_token=1024,
+        gpt_stop_audio_token=1025,
+        gpt_use_masking_gt_prompt_approach=True,
+        gpt_use_perceiver_resampler=True,
+    )
+    # define audio config
+    audio_config = XttsAudioConfig(sample_rate=22050, dvae_sample_rate=22050, output_sample_rate=24000)
+    # training parameters config
+
+    config = GPTTrainerConfig()
+
+    config.load_json(XTTS_CONFIG_FILE)
+
+    config.epochs = num_epochs
+    config.output_path = OUT_PATH
+    config.model_args = model_args
+    config.run_name = RUN_NAME
+    config.project_name = PROJECT_NAME
+    config.run_description = """
+        GPT XTTS training
+        """,
+    config.dashboard_logger = DASHBOARD_LOGGER
+    config.logger_uri = LOGGER_URI
+    config.audio = audio_config
+    config.batch_size = BATCH_SIZE
+    config.num_loader_workers = 8
+    config.eval_split_max_size = 256
+    config.print_step = 50
+    config.plot_step = 100
+    config.log_model_step = 100
+    config.save_n_checkpoints = 1       
+    config.save_checkpoints = False
+    config.print_eval = False
+    config.optimizer = "AdamW"
+    config.optimizer_wd_only_on_weights = OPTIMIZER_WD_ONLY_ON_WEIGHTS
+    config.optimizer_params = {"betas": [0.9, 0.96], "eps": 1e-8, "weight_decay": weight_decay}
+    config.lr = lr
+    config.lr_scheduler = "MultiStepLR"
+    config.lr_scheduler_params = {"milestones": [50000 * 18, 150000 * 18, 300000 * 18], "gamma": 0.5, "last_epoch": -1}
+    config.test_sentences = []
+    config.save_step = 9999999999
+
+
+    # init the model from config
+    model = GPTTrainer.init_from_config(config)
+
+    # load training samples
+    train_samples, eval_samples = load_tts_samples(
+        DATASETS_CONFIG_LIST,
+        eval_split=True,
+        eval_split_max_size=config.eval_split_max_size,
+        eval_split_size=config.eval_split_size,
+    )
+
+    # init the trainer and 🚀
+    trainer = Trainer(
+        TrainerArgs(
+            restore_path=None,  # xtts checkpoint is restored via xtts_checkpoint key so no need of restore it using Trainer restore_path parameter
+            skip_train_epoch=False,
+            start_with_eval=START_WITH_EVAL,
+            grad_accum_steps=GRAD_ACUMM_STEPS
+        ),
+        config,
+        output_path=os.path.join(output_path, "run", "training"),
+        model=model,
+        train_samples=train_samples,
+        eval_samples=eval_samples,
+    )
+    trainer.fit()
+    trainer_out_path = trainer.output_path
+
+    # deallocate VRAM and RAM
+    del model, trainer, train_samples, eval_samples
+    gc.collect()
+
+    return trainer_out_path
+
+if __name__ == "__main__":
+    parser = create_xtts_trainer_parser()
+    args = parser.parse_args()
+
+    trainer_out_path = train_gpt(
+        metadatas=args.metadatas,
+        output_path=args.output_path,
+        num_epochs=args.num_epochs,
+        batch_size=args.batch_size,
+        grad_acumm=args.grad_acumm,
+        weight_decay=args.weight_decay,
+        lr=args.lr,
+        max_text_length=args.max_text_length,
+        max_audio_length=args.max_audio_length,
+        save_step=args.save_step
+    )
+
+    print(f"Checkpoint saved in dir: {trainer_out_path}")

trainer_0_log.txt

@@ -0,0 +1,2135 @@
+ > Training Environment:
+ | > Backend: Torch
+ | > Mixed precision: False
+ | > Precision: float32
+ | > Current device: 0
+ | > Num. of GPUs: 1
+ | > Num. of CPUs: 4
+ | > Num. of Torch Threads: 1
+ | > Torch seed: 54321
+ | > Torch CUDNN: True
+ | > Torch CUDNN deterministic: False
+ | > Torch CUDNN benchmark: False
+ | > Torch TF32 MatMul: False
+ > Start Tensorboard: tensorboard --logdir=xTTS/checkpoints/GPT_XTTS_FT-June-02-2025_12+22PM-0000000
+
+ > Model has 521144678 parameters
+
+ > EPOCH: 0/2
+ --> xTTS/checkpoints/GPT_XTTS_FT-June-02-2025_12+22PM-0000000
+
+ > TRAINING (2025-06-02 12:22:56) 
+
+   --> TIME: 2025-06-02 12:23:01 -- STEP: 0/5792 -- GLOBAL_STEP: 0
+     | > loss_text_ce: 0.23782861232757568  (0.23782861232757568)
+     | > loss_mel_ce: 4.941317081451416  (4.941317081451416)
+     | > loss: 1.2947864532470703  (1.2947864532470703)
+     | > current_lr: 5e-06 
+     | > step_time: 1.3843  (1.3842837810516357)
+     | > loader_time: 4.3003  (4.3003199100494385)
+
+
+   --> TIME: 2025-06-02 12:23:30 -- STEP: 50/5792 -- GLOBAL_STEP: 50
+     | > loss_text_ce: 0.23064814507961273  (0.24020873457193376)
+     | > loss_mel_ce: 4.853140830993652  (4.867191286087037)
+     | > loss: 1.2709472179412842  (1.2768500065803527)
+     | > current_lr: 5e-06 
+     | > step_time: 0.2303  (0.3325099849700927)
+     | > loader_time: 0.0143  (0.016327280998229988)
+
+
+   --> TIME: 2025-06-02 12:23:59 -- STEP: 100/5792 -- GLOBAL_STEP: 100
+     | > loss_text_ce: 0.2179468423128128  (0.2335291735827923)
+     | > loss_mel_ce: 4.29183292388916  (4.678085584640503)
+     | > loss: 1.1274449825286865  (1.2279036939144128)
+     | > current_lr: 5e-06 
+     | > step_time: 0.2657  (0.3300197696685791)
+     | > loader_time: 0.0095  (0.015608358383178715)
+
+
+   --> TIME: 2025-06-02 12:24:28 -- STEP: 150/5792 -- GLOBAL_STEP: 150
+     | > loss_text_ce: 0.21278369426727295  (0.22793374240398406)
+     | > loss_mel_ce: 4.24708890914917  (4.557877604166666)
+     | > loss: 1.114968180656433  (1.1964528389771778)
+     | > current_lr: 5e-06 
+     | > step_time: 0.3034  (0.33196935812632256)
+     | > loader_time: 0.0147  (0.0150977897644043)
+
+
+   --> TIME: 2025-06-02 12:24:57 -- STEP: 200/5792 -- GLOBAL_STEP: 200
+     | > loss_text_ce: 0.21909523010253906  (0.22294333018362522)
+     | > loss_mel_ce: 3.632870674133301  (4.465021697282792)
+     | > loss: 0.96299147605896  (1.1719912576675409)
+     | > current_lr: 5e-06 
+     | > step_time: 0.137  (0.33124068975448623)
+     | > loader_time: 0.0134  (0.014897105693817142)
+
+
+   --> TIME: 2025-06-02 12:25:28 -- STEP: 250/5792 -- GLOBAL_STEP: 250
+     | > loss_text_ce: 0.20100712776184082  (0.21740962487459184)
+     | > loss_mel_ce: 4.115545749664307  (4.400661633491516)
+     | > loss: 1.0791382789611816  (1.154517814159393)
+     | > current_lr: 5e-06 
+     | > step_time: 0.4054  (0.33776610374450694)
+     | > loader_time: 0.0126  (0.014648478507995607)
+
+
+   --> TIME: 2025-06-02 12:25:57 -- STEP: 300/5792 -- GLOBAL_STEP: 300
+     | > loss_text_ce: 0.1881532222032547  (0.21308185448249173)
+     | > loss_mel_ce: 4.062322616577148  (4.338319851557412)
+     | > loss: 1.0626189708709717  (1.1378504260381062)
+     | > current_lr: 5e-06 
+     | > step_time: 0.3084  (0.33573377132415777)
+     | > loader_time: 0.0168  (0.014551755587259931)
+
+
+   --> TIME: 2025-06-02 12:26:25 -- STEP: 350/5792 -- GLOBAL_STEP: 350
+     | > loss_text_ce: 0.18258750438690186  (0.20880227374179014)
+     | > loss_mel_ce: 3.5383872985839844  (4.286162930216108)
+     | > loss: 0.930243730545044  (1.1237413001060488)
+     | > current_lr: 5e-06 
+     | > step_time: 0.2231  (0.3339259896959578)
+     | > loader_time: 0.0132  (0.014371032714843752)
+
+
+   --> TIME: 2025-06-02 12:26:54 -- STEP: 400/5792 -- GLOBAL_STEP: 400
+     | > loss_text_ce: 0.17027834057807922  (0.20482054084539406)
+     | > loss_mel_ce: 3.622598648071289  (4.2505981731414835)
+     | > loss: 0.9482192397117615  (1.113854677826166)
+     | > current_lr: 5e-06 
+     | > step_time: 0.3104  (0.3339052683115007)
+     | > loader_time: 0.0161  (0.014224189519882205)
+
+
+   --> TIME: 2025-06-02 12:27:23 -- STEP: 450/5792 -- GLOBAL_STEP: 450
+     | > loss_text_ce: 0.1692354679107666  (0.20091567562686066)
+     | > loss_mel_ce: 3.857461452484131  (4.213876393636072)
+     | > loss: 1.0066742897033691  (1.1036980164051056)
+     | > current_lr: 5e-06 
+     | > step_time: 0.3746  (0.33324571874406617)
+     | > loader_time: 0.0117  (0.014043218294779461)
+
+
+   --> TIME: 2025-06-02 12:27:52 -- STEP: 500/5792 -- GLOBAL_STEP: 500
+     | > loss_text_ce: 0.15726038813591003  (0.19705737382173533)
+     | > loss_mel_ce: 3.882636547088623  (4.188746206283573)
+     | > loss: 1.0099742412567139  (1.0964508949518204)
+     | > current_lr: 5e-06 
+     | > step_time: 0.2675  (0.3323551220893861)
+     | > loader_time: 0.0094  (0.01385477685928345)
+
+
+   --> TIME: 2025-06-02 12:28:21 -- STEP: 550/5792 -- GLOBAL_STEP: 550
+     | > loss_text_ce: 0.14702577888965607  (0.1931881040605631)
+     | > loss_mel_ce: 3.913630723953247  (4.1642507709156416)
+     | > loss: 1.0151641368865967  (1.0893597186695445)
+     | > current_lr: 5e-06 
+     | > step_time: 0.4815  (0.33244910023429186)
+     | > loader_time: 0.0126  (0.013701290217312902)
+
+
+   --> TIME: 2025-06-02 12:28:51 -- STEP: 600/5792 -- GLOBAL_STEP: 600
+     | > loss_text_ce: 0.13721345365047455  (0.18944158929089705)
+     | > loss_mel_ce: 4.101404190063477  (4.144896006981534)
+     | > loss: 1.0596543550491333  (1.0835843993226688)
+     | > current_lr: 5e-06 
+     | > step_time: 0.4056  (0.3338357198238373)
+     | > loader_time: 0.0101  (0.013622902234395351)
+
+
+   --> TIME: 2025-06-02 12:29:20 -- STEP: 650/5792 -- GLOBAL_STEP: 650
+     | > loss_text_ce: 0.142731174826622  (0.18570196391298224)
+     | > loss_mel_ce: 3.8508477210998535  (4.123029582683858)
+     | > loss: 0.9983947277069092  (1.0771828870589912)
+     | > current_lr: 5e-06 
+     | > step_time: 0.2568  (0.33367859180156995)
+     | > loader_time: 0.0125  (0.01352635603684646)
+
+
+   --> TIME: 2025-06-02 12:29:47 -- STEP: 700/5792 -- GLOBAL_STEP: 700
+     | > loss_text_ce: 0.13017334043979645  (0.18195634870656907)
+     | > loss_mel_ce: 3.6967780590057373  (4.096020159721375)
+     | > loss: 0.9567378759384155  (1.069494128142085)
+     | > current_lr: 5e-06 
+     | > step_time: 0.2121  (0.3316369649342126)
+     | > loader_time: 0.0094  (0.0133876953806196)
+
+
+   --> TIME: 2025-06-02 12:30:16 -- STEP: 750/5792 -- GLOBAL_STEP: 750
+     | > loss_text_ce: 0.1326562911272049  (0.17836018381516147)
+     | > loss_mel_ce: 3.863213300704956  (4.077398623466496)
+     | > loss: 0.9989674091339111  (1.0639397028287254)
+     | > current_lr: 5e-06 
+     | > step_time: 0.3401  (0.33136590449015263)
+     | > loader_time: 0.0089  (0.013331648508707687)
+
+
+   --> TIME: 2025-06-02 12:30:45 -- STEP: 800/5792 -- GLOBAL_STEP: 800
+     | > loss_text_ce: 0.11892132461071014  (0.17485223358497032)
+     | > loss_mel_ce: 3.5116255283355713  (4.059487291574481)
+     | > loss: 0.9076367020606995  (1.0585848820954558)
+     | > current_lr: 5e-06 
+     | > step_time: 0.1597  (0.331714543998241)
+     | > loader_time: 0.0106  (0.013222561478614812)
+
+
+   --> TIME: 2025-06-02 12:31:14 -- STEP: 850/5792 -- GLOBAL_STEP: 850
+     | > loss_text_ce: 0.11038005352020264  (0.1714906403685318)
+     | > loss_mel_ce: 3.8610212802886963  (4.046719538183775)
+     | > loss: 0.9928503036499023  (1.0545525454072382)
+     | > current_lr: 5e-06 
+     | > step_time: 0.3977  (0.3319235268761126)
+     | > loader_time: 0.012  (0.013164884623359236)
+
+
+   --> TIME: 2025-06-02 12:31:42 -- STEP: 900/5792 -- GLOBAL_STEP: 900
+     | > loss_text_ce: 0.11592362821102142  (0.1682853105333117)
+     | > loss_mel_ce: 4.2247114181518555  (4.031507659223343)
+     | > loss: 1.0851587057113647  (1.0499482430352092)
+     | > current_lr: 5e-06 
+     | > step_time: 0.3718  (0.3308940598699778)
+     | > loader_time: 0.0134  (0.01308101230197483)
+
+
+   --> TIME: 2025-06-02 12:32:10 -- STEP: 950/5792 -- GLOBAL_STEP: 950
+     | > loss_text_ce: 0.10771063715219498  (0.16517683673061828)
+     | > loss_mel_ce: 3.5032119750976562  (4.01872667036558)
+     | > loss: 0.9027306437492371  (1.045975877674001)
+     | > current_lr: 5e-06 
+     | > step_time: 0.1779  (0.33019118836051503)
+     | > loader_time: 0.0148  (0.01302782359876131)
+
+
+   --> TIME: 2025-06-02 12:32:38 -- STEP: 1000/5792 -- GLOBAL_STEP: 1000
+     | > loss_text_ce: 0.12572963535785675  (0.16233218172937638)
+     | > loss_mel_ce: 3.0970044136047363  (4.003747127771376)
+     | > loss: 0.8056834936141968  (1.041519828259944)
+     | > current_lr: 5e-06 
+     | > step_time: 0.1609  (0.32947225642204253)
+     | > loader_time: 0.0112  (0.01298054933547974)
+
+
+   --> TIME: 2025-06-02 12:33:07 -- STEP: 1050/5792 -- GLOBAL_STEP: 1050
+     | > loss_text_ce: 0.10119885206222534  (0.1596139285252208)
+     | > loss_mel_ce: 3.9164676666259766  (3.993313400404794)
+     | > loss: 1.004416584968567  (1.0382318332649403)
+     | > current_lr: 5e-06 
+     | > step_time: 0.4415  (0.33021083627428294)
+     | > loader_time: 0.0149  (0.01293485868544806)
+
+
+   --> TIME: 2025-06-02 12:33:36 -- STEP: 1100/5792 -- GLOBAL_STEP: 1100
+     | > loss_text_ce: 0.09815080463886261  (0.1569338711080226)
+     | > loss_mel_ce: 3.761666774749756  (3.9810155831683764)
+     | > loss: 0.9649543762207031  (1.0344873643463295)
+     | > current_lr: 5e-06 
+     | > step_time: 0.2093  (0.33017331686886836)
+     | > loader_time: 0.0095  (0.012836727445775815)
+
+
+   --> TIME: 2025-06-02 12:34:03 -- STEP: 1150/5792 -- GLOBAL_STEP: 1150
+     | > loss_text_ce: 0.09583915770053864  (0.15432746619634008)
+     | > loss_mel_ce: 3.7277543544769287  (3.967495829955391)
+     | > loss: 0.9558984041213989  (1.0304558247586937)
+     | > current_lr: 5e-06 
+     | > step_time: 0.5857  (0.3293305108858188)
+     | > loader_time: 0.0121  (0.01275255389835524)
+
+
+   --> TIME: 2025-06-02 12:34:32 -- STEP: 1200/5792 -- GLOBAL_STEP: 1200
+     | > loss_text_ce: 0.09167931973934174  (0.15185882307589063)
+     | > loss_mel_ce: 3.893519163131714  (3.9543634712696076)
+     | > loss: 0.9962996244430542  (1.0265555742382992)
+     | > current_lr: 5e-06 
+     | > step_time: 0.2053  (0.32926277975241314)
+     | > loader_time: 0.0085  (0.01267102936903636)
+
+
+   --> TIME: 2025-06-02 12:35:03 -- STEP: 1250/5792 -- GLOBAL_STEP: 1250
+     | > loss_text_ce: 0.09783565998077393  (0.14956970821619048)
+     | > loss_mel_ce: 3.702324867248535  (3.949050336265564)
+     | > loss: 0.9500401020050049  (1.024655011558532)
+     | > current_lr: 5e-06 
+     | > step_time: 0.4952  (0.3306157491683954)
+     | > loader_time: 0.0104  (0.012619953727722172)
+
+
+   --> TIME: 2025-06-02 12:35:31 -- STEP: 1300/5792 -- GLOBAL_STEP: 1300
+     | > loss_text_ce: 0.08431262522935867  (0.14732415511057936)
+     | > loss_mel_ce: 3.927443027496338  (3.9411918537433333)
+     | > loss: 1.0029388666152954  (1.0221290025344256)
+     | > current_lr: 5e-06 
+     | > step_time: 0.2377  (0.3301881036391619)
+     | > loader_time: 0.0087  (0.012530116668114296)
+
+
+   --> TIME: 2025-06-02 12:36:01 -- STEP: 1350/5792 -- GLOBAL_STEP: 1350
+     | > loss_text_ce: 0.08895734697580338  (0.14518685861870098)
+     | > loss_mel_ce: 3.606088638305664  (3.93310096723062)
+     | > loss: 0.9237614870071411  (1.0195719568376187)
+     | > current_lr: 5e-06 
+     | > step_time: 0.5945  (0.33128830627158784)
+     | > loader_time: 0.0219  (0.012501915649131492)
+
+
+   --> TIME: 2025-06-02 12:36:30 -- STEP: 1400/5792 -- GLOBAL_STEP: 1400
+     | > loss_text_ce: 0.08065199851989746  (0.14312162278486157)
+     | > loss_mel_ce: 3.0727732181549072  (3.924280370473862)
+     | > loss: 0.7883563041687012  (1.0168504986166955)
+     | > current_lr: 5e-06 
+     | > step_time: 0.1569  (0.3313496242250711)
+     | > loader_time: 0.0119  (0.012489057098116194)
+
+
+   --> TIME: 2025-06-02 12:37:00 -- STEP: 1450/5792 -- GLOBAL_STEP: 1450
+     | > loss_text_ce: 0.079571932554245  (0.14114773946075607)
+     | > loss_mel_ce: 3.952150821685791  (3.91605803686997)
+     | > loss: 1.0079306364059448  (1.0143014442509617)
+     | > current_lr: 5e-06 
+     | > step_time: 0.3056  (0.33169992216702143)
+     | > loader_time: 0.0124  (0.012441920247571221)
+
+
+   --> TIME: 2025-06-02 12:37:29 -- STEP: 1500/5792 -- GLOBAL_STEP: 1500
+     | > loss_text_ce: 0.08127571642398834  (0.1392633452763161)
+     | > loss_mel_ce: 3.6380221843719482  (3.905871781508128)
+     | > loss: 0.9298244714736938  (1.0112837819655731)
+     | > current_lr: 5e-06 
+     | > step_time: 0.2384  (0.33202595345179226)
+     | > loader_time: 0.0098  (0.012403525034586589)
+
+
+   --> TIME: 2025-06-02 12:37:59 -- STEP: 1550/5792 -- GLOBAL_STEP: 1550
+     | > loss_text_ce: 0.08357858657836914  (0.13743022495700477)
+     | > loss_mel_ce: 3.2771170139312744  (3.898509816354321)
+     | > loss: 0.8401739001274109  (1.0089850104624223)
+     | > current_lr: 5e-06 
+     | > step_time: 0.1768  (0.33223780908892203)
+     | > loader_time: 0.0085  (0.012367438347108902)
+
+
+   --> TIME: 2025-06-02 12:38:28 -- STEP: 1600/5792 -- GLOBAL_STEP: 1600
+     | > loss_text_ce: 0.07955673336982727  (0.13564207764342429)
+     | > loss_mel_ce: 3.7596960067749023  (3.892578792124987)
+     | > loss: 0.9598131775856018  (1.007055217660963)
+     | > current_lr: 5e-06 
+     | > step_time: 0.3524  (0.33262646302580806)
+     | > loader_time: 0.0101  (0.012325592041015625)
+
+
+   --> TIME: 2025-06-02 12:38:57 -- STEP: 1650/5792 -- GLOBAL_STEP: 1650
+     | > loss_text_ce: 0.08837619423866272  (0.13392942287253612)
+     | > loss_mel_ce: 3.777754068374634  (3.885421214681683)
+     | > loss: 0.9665325880050659  (1.0048376595973967)
+     | > current_lr: 5e-06 
+     | > step_time: 0.3983  (0.3327004941304522)
+     | > loader_time: 0.0101  (0.012279371637286561)
+
+
+   --> TIME: 2025-06-02 12:39:25 -- STEP: 1700/5792 -- GLOBAL_STEP: 1700
+     | > loss_text_ce: 0.07985109090805054  (0.13228617378455745)
+     | > loss_mel_ce: 3.664367198944092  (3.8788135032092823)
+     | > loss: 0.9360545873641968  (1.0027749196922073)
+     | > current_lr: 5e-06 
+     | > step_time: 0.1903  (0.33247794515946305)
+     | > loader_time: 0.0099  (0.012235631241517908)
+
+
+   --> TIME: 2025-06-02 12:39:53 -- STEP: 1750/5792 -- GLOBAL_STEP: 1750
+     | > loss_text_ce: 0.07385779917240143  (0.13067971873496242)
+     | > loss_mel_ce: 3.792656421661377  (3.8709569222586495)
+     | > loss: 0.9666285514831543  (1.0004091606821328)
+     | > current_lr: 5e-06 
+     | > step_time: 0.2969  (0.3322374545506066)
+     | > loader_time: 0.0094  (0.012205468041556223)
+
+
+   --> TIME: 2025-06-02 12:40:23 -- STEP: 1800/5792 -- GLOBAL_STEP: 1800
+     | > loss_text_ce: 0.07535833865404129  (0.12911513355664064)
+     | > loss_mel_ce: 3.611762046813965  (3.8656086026297674)
+     | > loss: 0.9217801094055176  (0.9986809344093001)
+     | > current_lr: 5e-06 
+     | > step_time: 0.4042  (0.3324654255972966)
+     | > loader_time: 0.014  (0.012192738983366225)
+
+
+   --> TIME: 2025-06-02 12:40:51 -- STEP: 1850/5792 -- GLOBAL_STEP: 1850
+     | > loss_text_ce: 0.0778559073805809  (0.12761117409814027)
+     | > loss_mel_ce: 3.0250251293182373  (3.857865214734464)
+     | > loss: 0.7757202386856079  (0.9963690974583493)
+     | > current_lr: 5e-06 
+     | > step_time: 0.2565  (0.3320673637132384)
+     | > loader_time: 0.0119  (0.012154008891131426)
+
+
+   --> TIME: 2025-06-02 12:41:21 -- STEP: 1900/5792 -- GLOBAL_STEP: 1900
+     | > loss_text_ce: 0.07138916105031967  (0.12619867451096817)
+     | > loss_mel_ce: 3.775388717651367  (3.851701809858021)
+     | > loss: 0.9616944789886475  (0.9944751213726241)
+     | > current_lr: 5e-06 
+     | > step_time: 0.2679  (0.3323917353780644)
+     | > loader_time: 0.0108  (0.012138623438383404)
+
+
+   --> TIME: 2025-06-02 12:41:50 -- STEP: 1950/5792 -- GLOBAL_STEP: 1950
+     | > loss_text_ce: 0.06446552276611328  (0.12474528638980308)
+     | > loss_mel_ce: 3.478724956512451  (3.8443781879620675)
+     | > loss: 0.8857976198196411  (0.9922808688114847)
+     | > current_lr: 5e-06 
+     | > step_time: 0.267  (0.3323717740865852)
+     | > loader_time: 0.0102  (0.012119236114697578)
+
+
+   --> TIME: 2025-06-02 12:42:16 -- STEP: 2000/5792 -- GLOBAL_STEP: 2000
+     | > loss_text_ce: 0.07915454357862473  (0.12336141603440026)
+     | > loss_mel_ce: 3.8885855674743652  (3.8369900035858153)
+     | > loss: 0.9919350147247314  (0.9900878550410267)
+     | > current_lr: 5e-06 
+     | > step_time: 0.3543  (0.3317557517290113)
+     | > loader_time: 0.0126  (0.012084858655929566)
+
+
+   --> TIME: 2025-06-02 12:42:46 -- STEP: 2050/5792 -- GLOBAL_STEP: 2050
+     | > loss_text_ce: 0.06675997376441956  (0.1220419639494359)
+     | > loss_mel_ce: 3.372281312942505  (3.8299568820581205)
+     | > loss: 0.8597603440284729  (0.9879997116181906)
+     | > current_lr: 5e-06 
+     | > step_time: 0.3444  (0.3319971689363803)
+     | > loader_time: 0.0105  (0.01205671554658471)
+
+
+   --> TIME: 2025-06-02 12:43:14 -- STEP: 2100/5792 -- GLOBAL_STEP: 2100
+     | > loss_text_ce: 0.0644061341881752  (0.12074772182142432)
+     | > loss_mel_ce: 3.0573296546936035  (3.823917001769656)
+     | > loss: 0.7804339528083801  (0.9861661809682841)
+     | > current_lr: 5e-06 
+     | > step_time: 0.1591  (0.33194411834080995)
+     | > loader_time: 0.0112  (0.012042945112500873)
+
+
+   --> TIME: 2025-06-02 12:43:45 -- STEP: 2150/5792 -- GLOBAL_STEP: 2150
+     | > loss_text_ce: 0.0625600665807724  (0.11952382788062077)
+     | > loss_mel_ce: 3.1828041076660156  (3.819017451530279)
+     | > loss: 0.8113410472869873  (0.984635319903839)
+     | > current_lr: 5e-06 
+     | > step_time: 0.2213  (0.33232841746751635)
+     | > loader_time: 0.0153  (0.012043467122454977)
+
+
+   --> TIME: 2025-06-02 12:44:12 -- STEP: 2200/5792 -- GLOBAL_STEP: 2200
+     | > loss_text_ce: 0.07580503076314926  (0.11829185286367463)
+     | > loss_mel_ce: 3.471283435821533  (3.8122500800002705)
+     | > loss: 0.886772096157074  (0.9826354832540855)
+     | > current_lr: 5e-06 
+     | > step_time: 0.2548  (0.33183346889235754)
+     | > loader_time: 0.0097  (0.012037343978881839)
+
+
+   --> TIME: 2025-06-02 12:44:41 -- STEP: 2250/5792 -- GLOBAL_STEP: 2250
+     | > loss_text_ce: 0.06772749871015549  (0.11713893160720648)
+     | > loss_mel_ce: 3.4121651649475098  (3.8072973964479235)
+     | > loss: 0.8699731826782227  (0.9811090819570749)
+     | > current_lr: 5e-06 
+     | > step_time: 0.196  (0.33194303947024884)
+     | > loader_time: 0.0094  (0.012012425316704644)
+
+
+   --> TIME: 2025-06-02 12:45:10 -- STEP: 2300/5792 -- GLOBAL_STEP: 2300
+     | > loss_text_ce: 0.061648476868867874  (0.11599788975942377)
+     | > loss_mel_ce: 3.568927764892578  (3.802683669069539)
+     | > loss: 0.9076440334320068  (0.9796703896574348)
+     | > current_lr: 5e-06 
+     | > step_time: 0.3098  (0.33178884682448023)
+     | > loader_time: 0.0146  (0.012008188081824262)
+
+
+   --> TIME: 2025-06-02 12:45:40 -- STEP: 2350/5792 -- GLOBAL_STEP: 2350
+     | > loss_text_ce: 0.06095200777053833  (0.11489018085947679)
+     | > loss_mel_ce: 3.651190996170044  (3.798658301576655)
+     | > loss: 0.9280357360839844  (0.9783871206324147)
+     | > current_lr: 5e-06 
+     | > step_time: 0.34  (0.3323820060364746)
+     | > loader_time: 0.0123  (0.011982986064667397)
+
+
+   --> TIME: 2025-06-02 12:46:08 -- STEP: 2400/5792 -- GLOBAL_STEP: 2400
+     | > loss_text_ce: 0.06559105217456818  (0.11383588070515527)
+     | > loss_mel_ce: 3.4982616901397705  (3.793631868362427)
+     | > loss: 0.8909631967544556  (0.9768669372300303)
+     | > current_lr: 5e-06 
+     | > step_time: 0.3963  (0.3320956921577454)
+     | > loader_time: 0.015  (0.011968173583348597)
+
+
+   --> TIME: 2025-06-02 12:46:40 -- STEP: 2450/5792 -- GLOBAL_STEP: 2450
+     | > loss_text_ce: 0.06471642851829529  (0.11277475817471118)
+     | > loss_mel_ce: 3.9599180221557617  (3.790205652275864)
+     | > loss: 1.0061585903167725  (0.9757451025807123)
+     | > current_lr: 5e-06 
+     | > step_time: 0.43  (0.3328273964901357)
+     | > loader_time: 0.0128  (0.011948673481843914)
+
+
+   --> TIME: 2025-06-02 12:47:08 -- STEP: 2500/5792 -- GLOBAL_STEP: 2500
+     | > loss_text_ce: 0.058090440928936005  (0.11174730639308675)
+     | > loss_mel_ce: 3.680204153060913  (3.786430072784424)
+     | > loss: 0.9345736503601074  (0.9745443446636196)
+     | > current_lr: 5e-06 
+     | > step_time: 0.4192  (0.3327032661437986)
+     | > loader_time: 0.014  (0.011920587730407725)
+
+
+   --> TIME: 2025-06-02 12:47:37 -- STEP: 2550/5792 -- GLOBAL_STEP: 2550
+     | > loss_text_ce: 0.06345102936029434  (0.11076159270224602)
+     | > loss_mel_ce: 3.3983774185180664  (3.7819082406922884)
+     | > loss: 0.8654571175575256  (0.9731674582584228)
+     | > current_lr: 5e-06 
+     | > step_time: 0.3032  (0.3327015813191728)
+     | > loader_time: 0.0101  (0.01190789082471062)
+
+
+   --> TIME: 2025-06-02 12:48:06 -- STEP: 2600/5792 -- GLOBAL_STEP: 2600
+     | > loss_text_ce: 0.05978523939847946  (0.10978607040041899)
+     | > loss_mel_ce: 3.5986759662628174  (3.7784423384299646)
+     | > loss: 0.914615273475647  (0.9720571020704046)
+     | > current_lr: 5e-06 
+     | > step_time: 0.2067  (0.33279164598538297)
+     | > loader_time: 0.0095  (0.01188810458550087)
+
+
+   --> TIME: 2025-06-02 12:48:38 -- STEP: 2650/5792 -- GLOBAL_STEP: 2650
+     | > loss_text_ce: 0.058306336402893066  (0.1088496894923582)
+     | > loss_mel_ce: 2.9056618213653564  (3.7759509926022226)
+     | > loss: 0.7409920692443848  (0.9712001704045058)
+     | > current_lr: 5e-06 
+     | > step_time: 0.2274  (0.3335093621487883)
+     | > loader_time: 0.0143  (0.01186871078779113)
+
+
+   --> TIME: 2025-06-02 12:49:06 -- STEP: 2700/5792 -- GLOBAL_STEP: 2700
+     | > loss_text_ce: 0.06381174176931381  (0.1079658736467912)
+     | > loss_mel_ce: 3.7307090759277344  (3.7717624486817254)
+     | > loss: 0.9486302137374878  (0.9699320805293539)
+     | > current_lr: 5e-06 
+     | > step_time: 0.3863  (0.33329126207916787)
+     | > loader_time: 0.0102  (0.01184851937823826)
+
+
+   --> TIME: 2025-06-02 12:49:37 -- STEP: 2750/5792 -- GLOBAL_STEP: 2750
+     | > loss_text_ce: 0.06515434384346008  (0.1070680166618389)
+     | > loss_mel_ce: 3.7349181175231934  (3.76892153228413)
+     | > loss: 0.9500181078910828  (0.9689973871707913)
+     | > current_lr: 5e-06 
+     | > step_time: 0.1943  (0.33362000248648843)
+     | > loader_time: 0.0092  (0.011837096214294438)
+
+
+   --> TIME: 2025-06-02 12:50:06 -- STEP: 2800/5792 -- GLOBAL_STEP: 2800
+     | > loss_text_ce: 0.0626336857676506  (0.10621006238939493)
+     | > loss_mel_ce: 3.619314193725586  (3.7653382179566792)
+     | > loss: 0.9204869866371155  (0.9678870699873989)
+     | > current_lr: 5e-06 
+     | > step_time: 0.3534  (0.3337795458521157)
+     | > loader_time: 0.0102  (0.011815384626388556)
+
+
+   --> TIME: 2025-06-02 12:50:34 -- STEP: 2850/5792 -- GLOBAL_STEP: 2850
+     | > loss_text_ce: 0.05888611823320389  (0.10537111821273949)
+     | > loss_mel_ce: 3.1223511695861816  (3.763010348939059)
+     | > loss: 0.79530930519104  (0.9670953666536428)
+     | > current_lr: 5e-06 
+     | > step_time: 0.2558  (0.33350726955815335)
+     | > loader_time: 0.0118  (0.011799037080062068)
+
+
+   --> TIME: 2025-06-02 12:51:03 -- STEP: 2900/5792 -- GLOBAL_STEP: 2900
+     | > loss_text_ce: 0.0631546750664711  (0.1045509203009563)
+     | > loss_mel_ce: 3.269371271133423  (3.7583381495804624)
+     | > loss: 0.8331314921379089  (0.9657222673399692)
+     | > current_lr: 5e-06 
+     | > step_time: 0.2418  (0.33377023384488846)
+     | > loader_time: 0.0097  (0.011781094896382304)
+
+
+   --> TIME: 2025-06-02 12:51:33 -- STEP: 2950/5792 -- GLOBAL_STEP: 2950
+     | > loss_text_ce: 0.054032497107982635  (0.10374996635999706)
+     | > loss_mel_ce: 3.7166588306427  (3.754722292625298)
+     | > loss: 0.942672848701477  (0.9646180645894193)
+     | > current_lr: 5e-06 
+     | > step_time: 0.4931  (0.3338233034489517)
+     | > loader_time: 0.0146  (0.01176468857264115)
+
+
+   --> TIME: 2025-06-02 12:52:03 -- STEP: 3000/5792 -- GLOBAL_STEP: 3000
+     | > loss_text_ce: 0.04497094824910164  (0.10296552998696751)
+     | > loss_mel_ce: 3.8430159091949463  (3.7516442841688793)
+     | > loss: 0.9719967246055603  (0.9636524533629415)
+     | > current_lr: 5e-06 
+     | > step_time: 0.2344  (0.3341527958710988)
+     | > loader_time: 0.0093  (0.011756853898366297)
+
+
+   --> TIME: 2025-06-02 12:52:32 -- STEP: 3050/5792 -- GLOBAL_STEP: 3050
+     | > loss_text_ce: 0.05044661462306976  (0.10219378712480172)
+     | > loss_mel_ce: 3.3637678623199463  (3.7479482764103373)
+     | > loss: 0.8535535931587219  (0.962535515730498)
+     | > current_lr: 5e-06 
+     | > step_time: 0.3357  (0.33412718593097107)
+     | > loader_time: 0.0094  (0.011745000745429371)
+
+
+   --> TIME: 2025-06-02 12:53:01 -- STEP: 3100/5792 -- GLOBAL_STEP: 3100
+     | > loss_text_ce: 0.0588652640581131  (0.10145230594421574)
+     | > loss_mel_ce: 3.9438345432281494  (3.7447509037294697)
+     | > loss: 1.0006749629974365  (0.9615508023769622)
+     | > current_lr: 5e-06 
+     | > step_time: 0.2943  (0.3342037938487145)
+     | > loader_time: 0.0097  (0.011726075987662043)
+
+
+   --> TIME: 2025-06-02 12:53:30 -- STEP: 3150/5792 -- GLOBAL_STEP: 3150
+     | > loss_text_ce: 0.051031917333602905  (0.10072415719902691)
+     | > loss_mel_ce: 3.401136636734009  (3.7408774338828192)
+     | > loss: 0.8630421161651611  (0.9604003977207908)
+     | > current_lr: 5e-06 
+     | > step_time: 0.2545  (0.334141366822379)
+     | > loader_time: 0.0144  (0.011702652280292818)
+
+
+   --> TIME: 2025-06-02 12:53:57 -- STEP: 3200/5792 -- GLOBAL_STEP: 3200
+     | > loss_text_ce: 0.06074317917227745  (0.10001451563904981)
+     | > loss_mel_ce: 3.513090133666992  (3.7376667942106723)
+     | > loss: 0.8934583067893982  (0.9594203274324533)
+     | > current_lr: 5e-06 
+     | > step_time: 0.2398  (0.3338389614969492)
+     | > loader_time: 0.012  (0.011692634075880054)
+
+
+   --> TIME: 2025-06-02 12:54:24 -- STEP: 3250/5792 -- GLOBAL_STEP: 3250
+     | > loss_text_ce: 0.05713744834065437  (0.0993218669753807)
+     | > loss_mel_ce: 3.3079988956451416  (3.7339409486330473)
+     | > loss: 0.8412840962409973  (0.9583157039055454)
+     | > current_lr: 5e-06 
+     | > step_time: 0.2647  (0.3333787508010864)
+     | > loader_time: 0.01  (0.011673195472130414)
+
+
+   --> TIME: 2025-06-02 12:54:53 -- STEP: 3300/5792 -- GLOBAL_STEP: 3300
+     | > loss_text_ce: 0.049549344927072525  (0.09864641397401226)
+     | > loss_mel_ce: 3.5223379135131836  (3.731030137611158)
+     | > loss: 0.8929718136787415  (0.9574191379005256)
+     | > current_lr: 5e-06 
+     | > step_time: 0.2054  (0.33343946623079707)
+     | > loader_time: 0.0094  (0.011662811438242599)
+
+
+   --> TIME: 2025-06-02 12:55:21 -- STEP: 3350/5792 -- GLOBAL_STEP: 3350
+     | > loss_text_ce: 0.053567271679639816  (0.09798200751974499)
+     | > loss_mel_ce: 3.276874542236328  (3.7281440085795388)
+     | > loss: 0.8326104283332825  (0.9565315039656053)
+     | > current_lr: 5e-06 
+     | > step_time: 0.5258  (0.3332891092015736)
+     | > loader_time: 0.0106  (0.011652176344572612)
+
+
+   --> TIME: 2025-06-02 12:55:52 -- STEP: 3400/5792 -- GLOBAL_STEP: 3400
+     | > loss_text_ce: 0.05310482531785965  (0.09735800301765683)
+     | > loss_mel_ce: 3.462460517883301  (3.7253227573282577)
+     | > loss: 0.8788913488388062  (0.9556701900503212)
+     | > current_lr: 5e-06 
+     | > step_time: 0.1824  (0.3334693743902094)
+     | > loader_time: 0.0087  (0.011638146919362689)
+
+
+   --> TIME: 2025-06-02 12:56:21 -- STEP: 3450/5792 -- GLOBAL_STEP: 3450
+     | > loss_text_ce: 0.05677446722984314  (0.0967331422789804)
+     | > loss_mel_ce: 3.994664430618286  (3.722395037084386)
+     | > loss: 1.0128597021102905  (0.954782044852989)
+     | > current_lr: 5e-06 
+     | > step_time: 0.3058  (0.333492807927339)
+     | > loader_time: 0.0099  (0.011624124844868982)
+
+
+   --> TIME: 2025-06-02 12:56:51 -- STEP: 3500/5792 -- GLOBAL_STEP: 3500
+     | > loss_text_ce: 0.058832015842199326  (0.0961081453689506)
+     | > loss_mel_ce: 3.7743239402770996  (3.7200029888153074)
+     | > loss: 0.9582889676094055  (0.9540277835471287)
+     | > current_lr: 5e-06 
+     | > step_time: 0.4594  (0.3338103825705392)
+     | > loader_time: 0.0135  (0.011612459046500073)
+
+
+   --> TIME: 2025-06-02 12:57:20 -- STEP: 3550/5792 -- GLOBAL_STEP: 3550
+     | > loss_text_ce: 0.050712406635284424  (0.09550555716832747)
+     | > loss_mel_ce: 3.5286216735839844  (3.7171594685567935)
+     | > loss: 0.894833505153656  (0.9531662564546287)
+     | > current_lr: 5e-06 
+     | > step_time: 0.3831  (0.33380109598938845)
+     | > loader_time: 0.0088  (0.011606321603479524)
+
+
+   --> TIME: 2025-06-02 12:57:48 -- STEP: 3600/5792 -- GLOBAL_STEP: 3600
+     | > loss_text_ce: 0.0556088350713253  (0.09492813276644371)
+     | > loss_mel_ce: 3.395554780960083  (3.713761396010717)
+     | > loss: 0.8627908825874329  (0.9521723822421495)
+     | > current_lr: 5e-06 
+     | > step_time: 0.2089  (0.3337154185109668)
+     | > loader_time: 0.0126  (0.01158974859449599)
+
+
+   --> TIME: 2025-06-02 12:58:17 -- STEP: 3650/5792 -- GLOBAL_STEP: 3650
+     | > loss_text_ce: 0.05382274463772774  (0.0943567333778697)
+     | > loss_mel_ce: 3.8336167335510254  (3.7102085386563655)
+     | > loss: 0.971859872341156  (0.9511413181317994)
+     | > current_lr: 5e-06 
+     | > step_time: 0.3718  (0.33372791290283205)
+     | > loader_time: 0.0094  (0.011583965510538183)
+
+
+   --> TIME: 2025-06-02 12:58:46 -- STEP: 3700/5792 -- GLOBAL_STEP: 3700
+     | > loss_text_ce: 0.054553646594285965  (0.09378471502983882)
+     | > loss_mel_ce: 3.3460400104522705  (3.707569275224531)
+     | > loss: 0.8501484394073486  (0.9503384976612552)
+     | > current_lr: 5e-06 
+     | > step_time: 0.239  (0.333674537491154)
+     | > loader_time: 0.0098  (0.01157456649316324)
+
+
+   --> TIME: 2025-06-02 12:59:16 -- STEP: 3750/5792 -- GLOBAL_STEP: 3750
+     | > loss_text_ce: 0.04969848692417145  (0.0932255767772594)
+     | > loss_mel_ce: 3.773103713989258  (3.705821061897278)
+     | > loss: 0.9557005763053894  (0.9497616597811379)
+     | > current_lr: 5e-06 
+     | > step_time: 0.4401  (0.33399006907145184)
+     | > loader_time: 0.0105  (0.011564516321818038)
+
+
+   --> TIME: 2025-06-02 12:59:45 -- STEP: 3800/5792 -- GLOBAL_STEP: 3800
+     | > loss_text_ce: 0.05216987058520317  (0.09270269069232429)
+     | > loss_mel_ce: 4.144445419311523  (3.703308340373792)
+     | > loss: 1.0491538047790527  (0.9490027578567202)
+     | > current_lr: 5e-06 
+     | > step_time: 0.3106  (0.3339822284171456)
+     | > loader_time: 0.0101  (0.011551208119643365)
+
+
+   --> TIME: 2025-06-02 13:00:15 -- STEP: 3850/5792 -- GLOBAL_STEP: 3850
+     | > loss_text_ce: 0.051087573170661926  (0.09218463780527755)
+     | > loss_mel_ce: 3.5129377841949463  (3.7008065200780895)
+     | > loss: 0.891006350517273  (0.9482477895625223)
+     | > current_lr: 5e-06 
+     | > step_time: 0.4407  (0.33412671906607494)
+     | > loader_time: 0.0109  (0.011542596259674472)
+
+
+   --> TIME: 2025-06-02 13:00:46 -- STEP: 3900/5792 -- GLOBAL_STEP: 3900
+     | > loss_text_ce: 0.05810575187206268  (0.09166618832888501)
+     | > loss_mel_ce: 3.621946334838867  (3.698486832961058)
+     | > loss: 0.9200130105018616  (0.9475382554378262)
+     | > current_lr: 5e-06 
+     | > step_time: 0.3832  (0.3345049955906012)
+     | > loader_time: 0.0108  (0.011529551591628641)
+
+
+   --> TIME: 2025-06-02 13:01:15 -- STEP: 3950/5792 -- GLOBAL_STEP: 3950
+     | > loss_text_ce: 0.05320081114768982  (0.09115826475186428)
+     | > loss_mel_ce: 3.2360873222351074  (3.695133606451976)
+     | > loss: 0.8223220109939575  (0.946572967918613)
+     | > current_lr: 5e-06 
+     | > step_time: 0.2365  (0.33447549988951863)
+     | > loader_time: 0.0092  (0.011514630860920198)
+
+
+   --> TIME: 2025-06-02 13:01:44 -- STEP: 4000/5792 -- GLOBAL_STEP: 4000
+     | > loss_text_ce: 0.05098037049174309  (0.09065508700720956)
+     | > loss_mel_ce: 3.636732578277588  (3.6921278865337372)
+     | > loss: 0.9219282269477844  (0.9456957434713839)
+     | > current_lr: 5e-06 
+     | > step_time: 0.3267  (0.33452891433238985)
+     | > loader_time: 0.0102  (0.01150596791505814)
+
+
+   --> TIME: 2025-06-02 13:02:12 -- STEP: 4050/5792 -- GLOBAL_STEP: 4050
+     | > loss_text_ce: 0.0510944202542305  (0.09017181752640514)
+     | > loss_mel_ce: 3.851416826248169  (3.6895573917436013)
+     | > loss: 0.9756278395652771  (0.9449323024131632)
+     | > current_lr: 5e-06 
+     | > step_time: 0.3061  (0.33426993028617197)
+     | > loader_time: 0.0091  (0.01149895055794422)
+
+
+   --> TIME: 2025-06-02 13:02:41 -- STEP: 4100/5792 -- GLOBAL_STEP: 4100
+     | > loss_text_ce: 0.059732966125011444  (0.0897003380263723)
+     | > loss_mel_ce: 4.043025016784668  (3.6875316979245443)
+     | > loss: 1.0256894826889038  (0.9443080091040307)
+     | > current_lr: 5e-06 
+     | > step_time: 0.3286  (0.33428408331987336)
+     | > loader_time: 0.0132  (0.011492944345241642)
+
+
+   --> TIME: 2025-06-02 13:03:11 -- STEP: 4150/5792 -- GLOBAL_STEP: 4150
+     | > loss_text_ce: 0.046467892825603485  (0.08922771856817126)
+     | > loss_mel_ce: 3.8112823963165283  (3.6847478388590984)
+     | > loss: 0.9644375443458557  (0.9434938894840605)
+     | > current_lr: 5e-06 
+     | > step_time: 0.3452  (0.3344374258546944)
+     | > loader_time: 0.0125  (0.011488211528364437)
+
+
+   --> TIME: 2025-06-02 13:03:41 -- STEP: 4200/5792 -- GLOBAL_STEP: 4200
+     | > loss_text_ce: 0.05000675097107887  (0.08878737421174125)
+     | > loss_mel_ce: 3.3880319595336914  (3.6820667290119897)
+     | > loss: 0.8595097064971924  (0.9427135259338786)
+     | > current_lr: 5e-06 
+     | > step_time: 0.2624  (0.33464160896482914)
+     | > loader_time: 0.0116  (0.011479487986791706)
+
+
+   --> TIME: 2025-06-02 13:04:10 -- STEP: 4250/5792 -- GLOBAL_STEP: 4250
+     | > loss_text_ce: 0.0450843945145607  (0.08833793204146267)
+     | > loss_mel_ce: 3.3565142154693604  (3.6793924145418058)
+     | > loss: 0.8503996729850769  (0.9419325868522416)
+     | > current_lr: 5e-06 
+     | > step_time: 0.5804  (0.3347087397855871)
+     | > loader_time: 0.0107  (0.011472317415125237)
+
+
+   --> TIME: 2025-06-02 13:04:41 -- STEP: 4300/5792 -- GLOBAL_STEP: 4300
+     | > loss_text_ce: 0.04820011556148529  (0.08789586786960442)
+     | > loss_mel_ce: 3.48346209526062  (3.677155670066213)
+     | > loss: 0.8829155564308167  (0.9412628846806143)
+     | > current_lr: 5e-06 
+     | > step_time: 0.2393  (0.33497945025909787)
+     | > loader_time: 0.0126  (0.011467096805572514)
+
+
+   --> TIME: 2025-06-02 13:05:08 -- STEP: 4350/5792 -- GLOBAL_STEP: 4350
+     | > loss_text_ce: 0.05529556795954704  (0.08746406216474098)
+     | > loss_mel_ce: 3.5293896198272705  (3.674542185136642)
+     | > loss: 0.8961712718009949  (0.9405015619869881)
+     | > current_lr: 5e-06 
+     | > step_time: 0.3449  (0.3346871621033242)
+     | > loader_time: 0.0123  (0.011460946072107077)
+
+
+   --> TIME: 2025-06-02 13:05:40 -- STEP: 4400/5792 -- GLOBAL_STEP: 4400
+     | > loss_text_ce: 0.0525493361055851  (0.08705084911886261)
+     | > loss_mel_ce: 3.2084221839904785  (3.6720786437663175)
+     | > loss: 0.8152428865432739  (0.9397823734175065)
+     | > current_lr: 5e-06 
+     | > step_time: 0.2394  (0.3350362862240186)
+     | > loader_time: 0.0123  (0.011455193866382943)
+
+
+   --> TIME: 2025-06-02 13:06:08 -- STEP: 4450/5792 -- GLOBAL_STEP: 4450
+     | > loss_text_ce: 0.048889096826314926  (0.0866304319134254)
+     | > loss_mel_ce: 3.5298917293548584  (3.670149624642362)
+     | > loss: 0.8946952223777771  (0.9391950143455109)
+     | > current_lr: 5e-06 
+     | > step_time: 0.5265  (0.33503451711676124)
+     | > loader_time: 0.0111  (0.011446116265286221)
+
+
+   --> TIME: 2025-06-02 13:06:37 -- STEP: 4500/5792 -- GLOBAL_STEP: 4500
+     | > loss_text_ce: 0.05733109265565872  (0.08622805037432246)
+     | > loss_mel_ce: 3.3458251953125  (3.666845300144621)
+     | > loss: 0.8507890701293945  (0.9382683378458013)
+     | > current_lr: 5e-06 
+     | > step_time: 0.2417  (0.33490972476535413)
+     | > loader_time: 0.0092  (0.011438722981346978)
+
+
+   --> TIME: 2025-06-02 13:07:05 -- STEP: 4550/5792 -- GLOBAL_STEP: 4550
+     | > loss_text_ce: 0.042130474001169205  (0.08583367971437321)
+     | > loss_mel_ce: 3.6195030212402344  (3.664244420135416)
+     | > loss: 0.9154083728790283  (0.9375195251716352)
+     | > current_lr: 5e-06 
+     | > step_time: 0.492  (0.33481265240973224)
+     | > loader_time: 0.0104  (0.011426460035554658)
+
+
+   --> TIME: 2025-06-02 13:07:33 -- STEP: 4600/5792 -- GLOBAL_STEP: 4600
+     | > loss_text_ce: 0.0529852919280529  (0.08543671885748273)
+     | > loss_mel_ce: 3.0790507793426514  (3.6618555622515485)
+     | > loss: 0.7830089926719666  (0.9368230704898409)
+     | > current_lr: 5e-06 
+     | > step_time: 0.2388  (0.3346719070102861)
+     | > loader_time: 0.0106  (0.011421545173810875)
+
+
+   --> TIME: 2025-06-02 13:08:05 -- STEP: 4650/5792 -- GLOBAL_STEP: 4650
+     | > loss_text_ce: 0.047616079449653625  (0.08505546001497138)
+     | > loss_mel_ce: 3.2773044109344482  (3.659559770348253)
+     | > loss: 0.831230103969574  (0.9361538077938931)
+     | > current_lr: 5e-06 
+     | > step_time: 0.406  (0.3350888223032801)
+     | > loader_time: 0.0104  (0.011418454980337488)
+
+
+   --> TIME: 2025-06-02 13:08:33 -- STEP: 4700/5792 -- GLOBAL_STEP: 4700
+     | > loss_text_ce: 0.04613576456904411  (0.08467220809072895)
+     | > loss_mel_ce: 3.406162977218628  (3.657190910603139)
+     | > loss: 0.8630746603012085  (0.9354657798751862)
+     | > current_lr: 5e-06 
+     | > step_time: 0.2167  (0.33492715972535186)
+     | > loader_time: 0.0128  (0.011416076599283414)
+
+
+   --> TIME: 2025-06-02 13:09:03 -- STEP: 4750/5792 -- GLOBAL_STEP: 4750
+     | > loss_text_ce: 0.052280884236097336  (0.08429848386896287)
+     | > loss_mel_ce: 3.40238356590271  (3.655417844571567)
+     | > loss: 0.8636661171913147  (0.9349290823058068)
+     | > current_lr: 5e-06 
+     | > step_time: 0.2554  (0.3351128557104816)
+     | > loader_time: 0.0099  (0.011411911663256183)
+
+
+   --> TIME: 2025-06-02 13:09:32 -- STEP: 4800/5792 -- GLOBAL_STEP: 4800
+     | > loss_text_ce: 0.042021121829748154  (0.08392824066414815)
+     | > loss_mel_ce: 3.1501052379608154  (3.653419889857373)
+     | > loss: 0.7980315685272217  (0.9343370328471056)
+     | > current_lr: 5e-06 
+     | > step_time: 0.2071  (0.33511566052834235)
+     | > loader_time: 0.0091  (0.011409775366385762)
+
+
+   --> TIME: 2025-06-02 13:10:01 -- STEP: 4850/5792 -- GLOBAL_STEP: 4850
+     | > loss_text_ce: 0.05289650708436966  (0.08356028458122745)
+     | > loss_mel_ce: 3.2315640449523926  (3.6508835728143922)
+     | > loss: 0.8211151361465454  (0.9336109645710768)
+     | > current_lr: 5e-06 
+     | > step_time: 0.335  (0.3350185093437278)
+     | > loader_time: 0.0092  (0.011409686619473473)
+
+
+   --> TIME: 2025-06-02 13:10:29 -- STEP: 4900/5792 -- GLOBAL_STEP: 4900
+     | > loss_text_ce: 0.04954209551215172  (0.08320543044683895)
+     | > loss_mel_ce: 3.3401761054992676  (3.6483470648648804)
+     | > loss: 0.8474295735359192  (0.9328881240499254)
+     | > current_lr: 5e-06 
+     | > step_time: 0.245  (0.3349105890916322)
+     | > loader_time: 0.01  (0.011410678065552989)
+
+
+   --> TIME: 2025-06-02 13:11:00 -- STEP: 4950/5792 -- GLOBAL_STEP: 4950
+     | > loss_text_ce: 0.04523472115397453  (0.08286285736118316)
+     | > loss_mel_ce: 3.421116352081299  (3.646339290888624)
+     | > loss: 0.86658775806427  (0.9323005372586868)
+     | > current_lr: 5e-06 
+     | > step_time: 0.267  (0.33518396363113906)
+     | > loader_time: 0.0095  (0.011406980524159427)
+
+
+   --> TIME: 2025-06-02 13:11:30 -- STEP: 5000/5792 -- GLOBAL_STEP: 5000
+     | > loss_text_ce: 0.051856279373168945  (0.08252262841463104)
+     | > loss_mel_ce: 3.7696542739868164  (3.64465701274872)
+     | > loss: 0.9553776383399963  (0.931794910478591)
+     | > current_lr: 5e-06 
+     | > step_time: 0.4771  (0.3353376725673679)
+     | > loader_time: 0.0107  (0.011403836870193468)
+
+
+   --> TIME: 2025-06-02 13:12:00 -- STEP: 5050/5792 -- GLOBAL_STEP: 5050
+     | > loss_text_ce: 0.048051297664642334  (0.08218267903026988)
+     | > loss_mel_ce: 3.2416117191314697  (3.6419559119007388)
+     | > loss: 0.8224157691001892  (0.9310346478990981)
+     | > current_lr: 5e-06 
+     | > step_time: 0.2594  (0.33541135504694297)
+     | > loader_time: 0.0094  (0.011395886864992641)
+
+
+   --> TIME: 2025-06-02 13:12:30 -- STEP: 5100/5792 -- GLOBAL_STEP: 5100
+     | > loss_text_ce: 0.04729149118065834  (0.0818471597763256)
+     | > loss_mel_ce: 3.463068962097168  (3.6400194923083)
+     | > loss: 0.8775901198387146  (0.9304666631712624)
+     | > current_lr: 5e-06 
+     | > step_time: 0.2577  (0.3356010669352963)
+     | > loader_time: 0.0093  (0.011388069274378751)
+
+
+   --> TIME: 2025-06-02 13:12:59 -- STEP: 5150/5792 -- GLOBAL_STEP: 5150
+     | > loss_text_ce: 0.049754828214645386  (0.08151657056822972)
+     | > loss_mel_ce: 3.3020455837249756  (3.638027641333424)
+     | > loss: 0.8379501104354858  (0.9298860531408801)
+     | > current_lr: 5e-06 
+     | > step_time: 0.1726  (0.33566474298829063)
+     | > loader_time: 0.0116  (0.011384222206560154)
+
+
+   --> TIME: 2025-06-02 13:13:28 -- STEP: 5200/5792 -- GLOBAL_STEP: 5200
+     | > loss_text_ce: 0.05030031129717827  (0.08118891543852057)
+     | > loss_mel_ce: 3.18542742729187  (3.635307474182204)
+     | > loss: 0.8089319467544556  (0.9291240975948472)
+     | > current_lr: 5e-06 
+     | > step_time: 0.3433  (0.3355758514771101)
+     | > loader_time: 0.0094  (0.011372857231360218)
+
+
+   --> TIME: 2025-06-02 13:13:59 -- STEP: 5250/5792 -- GLOBAL_STEP: 5250
+     | > loss_text_ce: 0.05008910596370697  (0.08087093743681945)
+     | > loss_mel_ce: 3.088693857192993  (3.6334633124669407)
+     | > loss: 0.7846957445144653  (0.9285835626579457)
+     | > current_lr: 5e-06 
+     | > step_time: 0.3717  (0.33582283274332736)
+     | > loader_time: 0.0095  (0.011365627288818372)
+
+
+   --> TIME: 2025-06-02 13:14:29 -- STEP: 5300/5792 -- GLOBAL_STEP: 5300
+     | > loss_text_ce: 0.0471770241856575  (0.08056517972277057)
+     | > loss_mel_ce: 3.359286069869995  (3.6318132130155036)
+     | > loss: 0.8516157865524292  (0.9280945984035159)
+     | > current_lr: 5e-06 
+     | > step_time: 0.3564  (0.33594532296342744)
+     | > loader_time: 0.0105  (0.011357401227051363)
+
+
+   --> TIME: 2025-06-02 13:14:58 -- STEP: 5350/5792 -- GLOBAL_STEP: 5350
+     | > loss_text_ce: 0.047488246113061905  (0.08026272471123812)
+     | > loss_mel_ce: 3.0119431018829346  (3.6297121964659658)
+     | > loss: 0.764857828617096  (0.9274937304937942)
+     | > current_lr: 5e-06 
+     | > step_time: 0.2594  (0.33586654542762623)
+     | > loader_time: 0.0141  (0.011354769679987546)
+
+
+   --> TIME: 2025-06-02 13:15:27 -- STEP: 5400/5792 -- GLOBAL_STEP: 5400
+     | > loss_text_ce: 0.047491975128650665  (0.079949514229127)
+     | > loss_mel_ce: 3.455355405807495  (3.627370389611634)
+     | > loss: 0.8757118582725525  (0.9268299761966414)
+     | > current_lr: 5e-06 
+     | > step_time: 0.2422  (0.3358160152258706)
+     | > loader_time: 0.0092  (0.011353673890784925)
+
+
+   --> TIME: 2025-06-02 13:15:54 -- STEP: 5450/5792 -- GLOBAL_STEP: 5450
+     | > loss_text_ce: 0.046924345195293427  (0.07964956542562501)
+     | > loss_mel_ce: 3.266174554824829  (3.625191910944949)
+     | > loss: 0.8282747268676758  (0.9262103693200899)
+     | > current_lr: 5e-06 
+     | > step_time: 0.4257  (0.3356467504238875)
+     | > loader_time: 0.0097  (0.011349368707849337)
+
+
+   --> TIME: 2025-06-02 13:16:25 -- STEP: 5500/5792 -- GLOBAL_STEP: 5500
+     | > loss_text_ce: 0.039876487106084824  (0.07935999841446274)
+     | > loss_mel_ce: 3.372520685195923  (3.62339421679757)
+     | > loss: 0.8530992865562439  (0.9256885540377001)
+     | > current_lr: 5e-06 
+     | > step_time: 0.2624  (0.3357687810984538)
+     | > loader_time: 0.0098  (0.011344045855782236)
+
+
+   --> TIME: 2025-06-02 13:16:53 -- STEP: 5550/5792 -- GLOBAL_STEP: 5550
+     | > loss_text_ce: 0.05153404176235199  (0.07907796377802799)
+     | > loss_mel_ce: 3.4474453926086426  (3.6210933367411307)
+     | > loss: 0.8747448325157166  (0.9250428253680728)
+     | > current_lr: 5e-06 
+     | > step_time: 0.2201  (0.3357482298000451)
+     | > loader_time: 0.0085  (0.011339900300309447)
+
+
+   --> TIME: 2025-06-02 13:17:23 -- STEP: 5600/5792 -- GLOBAL_STEP: 5600
+     | > loss_text_ce: 0.05078967288136482  (0.07880074134562218)
+     | > loss_mel_ce: 3.4180071353912354  (3.6193902756060887)
+     | > loss: 0.867199182510376  (0.9245477544835627)
+     | > current_lr: 5e-06 
+     | > step_time: 0.2526  (0.33589044515575817)
+     | > loader_time: 0.0093  (0.011337815778596043)
+
+
+   --> TIME: 2025-06-02 13:17:51 -- STEP: 5650/5792 -- GLOBAL_STEP: 5650
+     | > loss_text_ce: 0.053694888949394226  (0.07851750885227134)
+     | > loss_mel_ce: 3.102844476699829  (3.6172207505724083)
+     | > loss: 0.7891348600387573  (0.9239345651073785)
+     | > current_lr: 5e-06 
+     | > step_time: 0.3004  (0.33572653774666705)
+     | > loader_time: 0.0096  (0.011334677755305183)
+
+
+   --> TIME: 2025-06-02 13:18:20 -- STEP: 5700/5792 -- GLOBAL_STEP: 5700
+     | > loss_text_ce: 0.0459514856338501  (0.07823847596974794)
+     | > loss_mel_ce: 3.5479061603546143  (3.615112853050233)
+     | > loss: 0.8984644412994385  (0.9233378325533441)
+     | > current_lr: 5e-06 
+     | > step_time: 0.3254  (0.3357153525687113)
+     | > loader_time: 0.0106  (0.011330316192225392)
+
+
+   --> TIME: 2025-06-02 13:18:49 -- STEP: 5750/5792 -- GLOBAL_STEP: 5750
+     | > loss_text_ce: 0.04507908225059509  (0.07796999881967298)
+     | > loss_mel_ce: 3.325566530227661  (3.613506336834121)
+     | > loss: 0.8426613807678223  (0.9228690842441882)
+     | > current_lr: 5e-06 
+     | > step_time: 0.2647  (0.3357090207804818)
+     | > loader_time: 0.0132  (0.011328426651332674)
+
+
+ > EVALUATION 
+
+
+  --> EVAL PERFORMANCE
+     | > avg_loader_time: 0.2016423004495222 (+0)
+     | > avg_loss_text_ce: 0.04477991773036578 (+0)
+     | > avg_loss_mel_ce: 3.199991709768435 (+0)
+     | > avg_loss: 3.2447716307505363 (+0)
+
+
+ > EPOCH: 1/2
+ --> xTTS/checkpoints/GPT_XTTS_FT-June-02-2025_12+22PM-0000000
+
+ > TRAINING (2025-06-02 13:26:46) 
+
+   --> TIME: 2025-06-02 13:26:52 -- STEP: 8/5792 -- GLOBAL_STEP: 5800
+     | > loss_text_ce: 0.04962650686502457  (0.04589929943904281)
+     | > loss_mel_ce: 3.476445436477661  (3.366139441728592)
+     | > loss: 0.8815180063247681  (0.8530096858739853)
+     | > current_lr: 5e-06 
+     | > step_time: 0.2652  (0.30354249477386475)
+     | > loader_time: 0.0149  (0.011465787887573242)
+
+
+   --> TIME: 2025-06-02 13:27:22 -- STEP: 58/5792 -- GLOBAL_STEP: 5850
+     | > loss_text_ce: 0.043131984770298004  (0.046852813262877796)
+     | > loss_mel_ce: 3.372774600982666  (3.4303042765321403)
+     | > loss: 0.8539766669273376  (0.8692892707627395)
+     | > current_lr: 5e-06 
+     | > step_time: 0.196  (0.3431942175174581)
+     | > loader_time: 0.0127  (0.01153468263560328)
+
+
+   --> TIME: 2025-06-02 13:27:52 -- STEP: 108/5792 -- GLOBAL_STEP: 5900
+     | > loss_text_ce: 0.04623452201485634  (0.04628198635247019)
+     | > loss_mel_ce: 2.833378553390503  (3.3930641324431807)
+     | > loss: 0.719903290271759  (0.8598365292504981)
+     | > current_lr: 5e-06 
+     | > step_time: 0.1939  (0.3511013057496811)
+     | > loader_time: 0.0098  (0.011563482107939544)
+
+
+   --> TIME: 2025-06-02 13:28:22 -- STEP: 158/5792 -- GLOBAL_STEP: 5950
+     | > loss_text_ce: 0.04444996267557144  (0.04651183649143084)
+     | > loss_mel_ce: 3.0688369274139404  (3.396478888354724)
+     | > loss: 0.7783217430114746  (0.8607476810111275)
+     | > current_lr: 5e-06 
+     | > step_time: 0.3297  (0.34766890278345425)
+     | > loader_time: 0.0103  (0.01131993305834034)
+
+
+   --> TIME: 2025-06-02 13:28:52 -- STEP: 208/5792 -- GLOBAL_STEP: 6000
+     | > loss_text_ce: 0.05267365276813507  (0.04660090401123921)
+     | > loss_mel_ce: 3.509939670562744  (3.395385509500137)
+     | > loss: 0.8906533122062683  (0.8604966015196763)
+     | > current_lr: 5e-06 
+     | > step_time: 0.2086  (0.34758124213952296)
+     | > loader_time: 0.0099  (0.01136978314473079)
+
+
+   --> TIME: 2025-06-02 13:29:21 -- STEP: 258/5792 -- GLOBAL_STEP: 6050
+     | > loss_text_ce: 0.042573895305395126  (0.04650543012129243)
+     | > loss_mel_ce: 3.198530673980713  (3.406408476275067)
+     | > loss: 0.8102761507034302  (0.8632284749385922)
+     | > current_lr: 5e-06 
+     | > step_time: 0.2678  (0.34636553313380997)
+     | > loader_time: 0.0093  (0.011229625968045968)
+
+
+   --> TIME: 2025-06-02 13:29:52 -- STEP: 308/5792 -- GLOBAL_STEP: 6100
+     | > loss_text_ce: 0.04070745408535004  (0.04644479884391093)
+     | > loss_mel_ce: 3.161195993423462  (3.409244562124277)
+     | > loss: 0.8004758358001709  (0.8639223389811327)
+     | > current_lr: 5e-06 
+     | > step_time: 0.2063  (0.34847222055707666)
+     | > loader_time: 0.0096  (0.0111765776361738)
+
+
+   --> TIME: 2025-06-02 13:30:21 -- STEP: 358/5792 -- GLOBAL_STEP: 6150
+     | > loss_text_ce: 0.04625903442502022  (0.04643882370902503)
+     | > loss_mel_ce: 3.097196578979492  (3.4112956583832896)
+     | > loss: 0.7858638763427734  (0.8644336197296332)
+     | > current_lr: 5e-06 
+     | > step_time: 0.2922  (0.3459169138743227)
+     | > loader_time: 0.01  (0.011164890321273387)
+
+
+   --> TIME: 2025-06-02 13:30:50 -- STEP: 408/5792 -- GLOBAL_STEP: 6200
+     | > loss_text_ce: 0.048108890652656555  (0.04640357185374285)
+     | > loss_mel_ce: 3.2346646785736084  (3.406046163217694)
+     | > loss: 0.8206933736801147  (0.8631124335761161)
+     | > current_lr: 5e-06 
+     | > step_time: 0.3294  (0.3447357386934993)
+     | > loader_time: 0.0094  (0.01113292983933992)
+
+
+   --> TIME: 2025-06-02 13:31:20 -- STEP: 458/5792 -- GLOBAL_STEP: 6250
+     | > loss_text_ce: 0.048285942524671555  (0.046448883360585246)
+     | > loss_mel_ce: 3.597545862197876  (3.4110943599559334)
+     | > loss: 0.9114579558372498  (0.8643858104814085)
+     | > current_lr: 5e-06 
+     | > step_time: 0.5765  (0.34515097307846554)
+     | > loader_time: 0.0139  (0.011161539752410497)
+
+
+   --> TIME: 2025-06-02 13:31:50 -- STEP: 508/5792 -- GLOBAL_STEP: 6300
+     | > loss_text_ce: 0.05213044583797455  (0.04643937362372638)
+     | > loss_mel_ce: 3.4266021251678467  (3.4056746645236577)
+     | > loss: 0.8696831464767456  (0.8630285095276793)
+     | > current_lr: 5e-06 
+     | > step_time: 0.3125  (0.3454259840522227)
+     | > loader_time: 0.0139  (0.011176920781924038)
+
+
+   --> TIME: 2025-06-02 13:32:18 -- STEP: 558/5792 -- GLOBAL_STEP: 6350
+     | > loss_text_ce: 0.04106414318084717  (0.046370733649499944)
+     | > loss_mel_ce: 3.4639627933502197  (3.397692242830885)
+     | > loss: 0.8762567043304443  (0.8610157443203804)
+     | > current_lr: 5e-06 
+     | > step_time: 0.1901  (0.3433396858981008)
+     | > loader_time: 0.0088  (0.011147387138831576)
+
+
+   --> TIME: 2025-06-02 13:32:45 -- STEP: 608/5792 -- GLOBAL_STEP: 6400
+     | > loss_text_ce: 0.04276609793305397  (0.04636223264969883)
+     | > loss_mel_ce: 3.2456772327423096  (3.394306463630576)
+     | > loss: 0.8221108317375183  (0.8601671744530135)
+     | > current_lr: 5e-06 
+     | > step_time: 0.2159  (0.3403207224450615)
+     | > loader_time: 0.0096  (0.011096751611483725)
+
+
+   --> TIME: 2025-06-02 13:33:15 -- STEP: 658/5792 -- GLOBAL_STEP: 6450
+     | > loss_text_ce: 0.04221029207110405  (0.046298709672544465)
+     | > loss_mel_ce: 3.376603364944458  (3.3917605297181384)
+     | > loss: 0.854703426361084  (0.8595148100316702)
+     | > current_lr: 5e-06 
+     | > step_time: 0.2959  (0.3403355002765601)
+     | > loader_time: 0.0123  (0.01107824609634724)
+
+
+   --> TIME: 2025-06-02 13:33:45 -- STEP: 708/5792 -- GLOBAL_STEP: 6500
+     | > loss_text_ce: 0.0455298125743866  (0.04620274585161333)
+     | > loss_mel_ce: 3.368074655532837  (3.3912849278099793)
+     | > loss: 0.8534011244773865  (0.8593719188271266)
+     | > current_lr: 5e-06 
+     | > step_time: 0.2983  (0.34120621048124494)
+     | > loader_time: 0.0097  (0.011063059507790253)
+
+
+   --> TIME: 2025-06-02 13:34:16 -- STEP: 758/5792 -- GLOBAL_STEP: 6550
+     | > loss_text_ce: 0.04287204146385193  (0.046180617649078645)
+     | > loss_mel_ce: 3.8676905632019043  (3.3931626326805056)
+     | > loss: 0.9776406288146973  (0.8598358130234839)
+     | > current_lr: 5e-06 
+     | > step_time: 0.4669  (0.3418154071692112)
+     | > loader_time: 0.0107  (0.011052280743078057)
+
+
+   --> TIME: 2025-06-02 13:34:44 -- STEP: 808/5792 -- GLOBAL_STEP: 6600
+     | > loss_text_ce: 0.05415653437376022  (0.046192656469411446)
+     | > loss_mel_ce: 3.0190370082855225  (3.3884078290792976)
+     | > loss: 0.7682983875274658  (0.8586501218747382)
+     | > current_lr: 5e-06 
+     | > step_time: 0.2155  (0.3408614199350376)
+     | > loader_time: 0.0131  (0.011081166196577612)
+
+
+   --> TIME: 2025-06-02 13:35:13 -- STEP: 858/5792 -- GLOBAL_STEP: 6650
+     | > loss_text_ce: 0.0525716096162796  (0.04613834280425154)
+     | > loss_mel_ce: 3.157501459121704  (3.3880747826227218)
+     | > loss: 0.8025182485580444  (0.8585532816834658)
+     | > current_lr: 5e-06 
+     | > step_time: 0.3441  (0.34018004690850534)
+     | > loader_time: 0.0116  (0.01113530639168265)
+
+
+   --> TIME: 2025-06-02 13:35:43 -- STEP: 908/5792 -- GLOBAL_STEP: 6700
+     | > loss_text_ce: 0.04388509690761566  (0.04611085562060055)
+     | > loss_mel_ce: 3.5000290870666504  (3.382610232031818)
+     | > loss: 0.8859785199165344  (0.8571802720361866)
+     | > current_lr: 5e-06 
+     | > step_time: 0.2732  (0.3402811882254311)
+     | > loader_time: 0.0098  (0.01113482561405534)
+
+
+   --> TIME: 2025-06-02 13:36:12 -- STEP: 958/5792 -- GLOBAL_STEP: 6750
+     | > loss_text_ce: 0.04200979322195053  (0.046016689794359696)
+     | > loss_mel_ce: 3.2025482654571533  (3.386244950812147)
+     | > loss: 0.8111395239830017  (0.858065410110796)
+     | > current_lr: 5e-06 
+     | > step_time: 0.3037  (0.3404792111701408)
+     | > loader_time: 0.013  (0.011161978409036457)
+
+
+   --> TIME: 2025-06-02 13:36:43 -- STEP: 1008/5792 -- GLOBAL_STEP: 6800
+     | > loss_text_ce: 0.04656248539686203  (0.045978479368227784)
+     | > loss_mel_ce: 3.424415349960327  (3.3894127905368805)
+     | > loss: 0.8677444458007812  (0.8588478174947555)
+     | > current_lr: 5e-06 
+     | > step_time: 0.2899  (0.34167367078009103)
+     | > loader_time: 0.0134  (0.011164178214375927)
+
+
+   --> TIME: 2025-06-02 13:37:12 -- STEP: 1058/5792 -- GLOBAL_STEP: 6850
+     | > loss_text_ce: 0.044847793877124786  (0.04588977389727736)
+     | > loss_mel_ce: 3.753209352493286  (3.3859440919807593)
+     | > loss: 0.9495142698287964  (0.8579584664342983)
+     | > current_lr: 5e-06 
+     | > step_time: 0.4274  (0.3409366860056194)
+     | > loader_time: 0.0106  (0.01115854092942293)
+
+
+   --> TIME: 2025-06-02 13:37:40 -- STEP: 1108/5792 -- GLOBAL_STEP: 6900
+     | > loss_text_ce: 0.04726967215538025  (0.04582478893443348)
+     | > loss_mel_ce: 3.5899248123168945  (3.3846601099313816)
+     | > loss: 0.9092985987663269  (0.8576212246172693)
+     | > current_lr: 5e-06 
+     | > step_time: 0.2437  (0.34045616211873964)
+     | > loader_time: 0.0129  (0.01116837792448188)
+
+
+   --> TIME: 2025-06-02 13:38:07 -- STEP: 1158/5792 -- GLOBAL_STEP: 6950
+     | > loss_text_ce: 0.04457278549671173  (0.04580180395864244)
+     | > loss_mel_ce: 3.8512332439422607  (3.3820696817984452)
+     | > loss: 0.973951518535614  (0.8569678713620632)
+     | > current_lr: 5e-06 
+     | > step_time: 0.4823  (0.33921163193302456)
+     | > loader_time: 0.0142  (0.011172776601063361)
+
+
+   --> TIME: 2025-06-02 13:38:35 -- STEP: 1208/5792 -- GLOBAL_STEP: 7000
+     | > loss_text_ce: 0.04959622770547867  (0.04575508566178527)
+     | > loss_mel_ce: 3.364575147628784  (3.3837944628387104)
+     | > loss: 0.8535428643226624  (0.8573873870518821)
+     | > current_lr: 5e-06 
+     | > step_time: 0.2847  (0.3381390390017176)
+     | > loader_time: 0.0111  (0.011142225849707395)
+
+
+   --> TIME: 2025-06-02 13:39:03 -- STEP: 1258/5792 -- GLOBAL_STEP: 7050
+     | > loss_text_ce: 0.04021007567644119  (0.04573283367309072)
+     | > loss_mel_ce: 3.0035552978515625  (3.3824539863058805)
+     | > loss: 0.7609413266181946  (0.857046704982157)
+     | > current_lr: 5e-06 
+     | > step_time: 0.2549  (0.33753872296768467)
+     | > loader_time: 0.0098  (0.011134225537553312)
+
+
+   --> TIME: 2025-06-02 13:39:33 -- STEP: 1308/5792 -- GLOBAL_STEP: 7100
+     | > loss_text_ce: 0.04839656129479408  (0.04570206533263035)
+     | > loss_mel_ce: 3.5808684825897217  (3.3811058704641623)
+     | > loss: 0.907316267490387  (0.8567019838623313)
+     | > current_lr: 5e-06 
+     | > step_time: 0.2648  (0.33785488281045833)
+     | > loader_time: 0.0126  (0.01116103261982629)
+
+
+   --> TIME: 2025-06-02 13:40:03 -- STEP: 1358/5792 -- GLOBAL_STEP: 7150
+     | > loss_text_ce: 0.04666044935584068  (0.04564322057743835)
+     | > loss_mel_ce: 3.6217894554138184  (3.380934479310341)
+     | > loss: 0.9171124696731567  (0.8566444249321679)
+     | > current_lr: 5e-06 
+     | > step_time: 0.3762  (0.33808904807241025)
+     | > loader_time: 0.0109  (0.011158927726464696)
+
+
+   --> TIME: 2025-06-02 13:40:32 -- STEP: 1408/5792 -- GLOBAL_STEP: 7200
+     | > loss_text_ce: 0.04733112081885338  (0.04561400014790704)
+     | > loss_mel_ce: 3.2079885005950928  (3.3785352163355467)
+     | > loss: 0.8138298988342285  (0.8560373041440138)
+     | > current_lr: 5e-06 
+     | > step_time: 0.3122  (0.33810166117142526)
+     | > loader_time: 0.0126  (0.01115033944899386)
+
+
+   --> TIME: 2025-06-02 13:41:01 -- STEP: 1458/5792 -- GLOBAL_STEP: 7250
+     | > loss_text_ce: 0.04578738287091255  (0.04558497288140411)
+     | > loss_mel_ce: 3.535698652267456  (3.3770085259229563)
+     | > loss: 0.8953714966773987  (0.8556483748109578)
+     | > current_lr: 5e-06 
+     | > step_time: 0.4838  (0.3378712419619776)
+     | > loader_time: 0.0114  (0.011147373661255806)
+
+
+   --> TIME: 2025-06-02 13:41:31 -- STEP: 1508/5792 -- GLOBAL_STEP: 7300
+     | > loss_text_ce: 0.04656374827027321  (0.0455558022918965)
+     | > loss_mel_ce: 3.0295422077178955  (3.3743492266859874)
+     | > loss: 0.769026517868042  (0.8549762572944953)
+     | > current_lr: 5e-06 
+     | > step_time: 0.1911  (0.3378510261720308)
+     | > loader_time: 0.0125  (0.011148763430213425)
+
+
+   --> TIME: 2025-06-02 13:42:00 -- STEP: 1558/5792 -- GLOBAL_STEP: 7350
+     | > loss_text_ce: 0.04521656408905983  (0.04553002290743304)
+     | > loss_mel_ce: 3.1920225620269775  (3.3758582573034954)
+     | > loss: 0.8093097805976868  (0.8553470702165203)
+     | > current_lr: 5e-06 
+     | > step_time: 0.2611  (0.33788349662114764)
+     | > loader_time: 0.0099  (0.011138344300726721)
+
+
+   --> TIME: 2025-06-02 13:42:28 -- STEP: 1608/5792 -- GLOBAL_STEP: 7400
+     | > loss_text_ce: 0.04438360407948494  (0.04550991758741251)
+     | > loss_mel_ce: 3.5675711631774902  (3.3738133661189487)
+     | > loss: 0.9029886722564697  (0.8548308210215757)
+     | > current_lr: 5e-06 
+     | > step_time: 0.2961  (0.33742392522778664)
+     | > loader_time: 0.0104  (0.011117274488382674)
+
+
+   --> TIME: 2025-06-02 13:42:59 -- STEP: 1658/5792 -- GLOBAL_STEP: 7450
+     | > loss_text_ce: 0.04301420599222183  (0.045442521491226136)
+     | > loss_mel_ce: 3.6960628032684326  (3.37488229901713)
+     | > loss: 0.9347692728042603  (0.8550812052371849)
+     | > current_lr: 5e-06 
+     | > step_time: 0.3433  (0.3380332888970184)
+     | > loader_time: 0.0102  (0.011103294151683483)
+
+
+   --> TIME: 2025-06-02 13:43:27 -- STEP: 1708/5792 -- GLOBAL_STEP: 7500
+     | > loss_text_ce: 0.043877944350242615  (0.0453859308027865)
+     | > loss_mel_ce: 2.9800140857696533  (3.373501475298433)
+     | > loss: 0.7559729814529419  (0.854721851594554)
+     | > current_lr: 5e-06 
+     | > step_time: 0.2149  (0.3374889004425923)
+     | > loader_time: 0.0099  (0.011108080732180309)
+
+
+   --> TIME: 2025-06-02 13:43:55 -- STEP: 1758/5792 -- GLOBAL_STEP: 7550
+     | > loss_text_ce: 0.03403538465499878  (0.04534231755878262)
+     | > loss_mel_ce: 2.907336711883545  (3.3728728446266083)
+     | > loss: 0.7353430390357971  (0.8545537905956154)
+     | > current_lr: 5e-06 
+     | > step_time: 0.2952  (0.33705969068377484)
+     | > loader_time: 0.0126  (0.011120306347010488)
+
+
+   --> TIME: 2025-06-02 13:44:24 -- STEP: 1808/5792 -- GLOBAL_STEP: 7600
+     | > loss_text_ce: 0.036792825907468796  (0.04528982559552794)
+     | > loss_mel_ce: 3.696239948272705  (3.3725063265424917)
+     | > loss: 0.9332581758499146  (0.8544490379569804)
+     | > current_lr: 5e-06 
+     | > step_time: 0.2613  (0.33688090253720226)
+     | > loader_time: 0.0099  (0.011115989721981826)
+
+
+   --> TIME: 2025-06-02 13:44:52 -- STEP: 1858/5792 -- GLOBAL_STEP: 7650
+     | > loss_text_ce: 0.0398641973733902  (0.04524051058893085)
+     | > loss_mel_ce: 3.4899098873138428  (3.3701283141280918)
+     | > loss: 0.8824435472488403  (0.8538422060025648)
+     | > current_lr: 5e-06 
+     | > step_time: 0.2554  (0.3364653570659706)
+     | > loader_time: 0.0106  (0.011119012812110403)
+
+
+   --> TIME: 2025-06-02 13:45:21 -- STEP: 1908/5792 -- GLOBAL_STEP: 7700
+     | > loss_text_ce: 0.041133489459753036  (0.04522936787179024)
+     | > loss_mel_ce: 3.302356481552124  (3.369759842034906)
+     | > loss: 0.83587247133255  (0.8537473023187688)
+     | > current_lr: 5e-06 
+     | > step_time: 0.2272  (0.33635687903038347)
+     | > loader_time: 0.0118  (0.011123857283242345)
+
+
+   --> TIME: 2025-06-02 13:45:49 -- STEP: 1958/5792 -- GLOBAL_STEP: 7750
+     | > loss_text_ce: 0.04508072882890701  (0.045202246298242925)
+     | > loss_mel_ce: 4.015277862548828  (3.3691428928988953)
+     | > loss: 1.0150896310806274  (0.8535862845712101)
+     | > current_lr: 5e-06 
+     | > step_time: 0.2578  (0.33611000692272075)
+     | > loader_time: 0.01  (0.01110899363158307)
+
+
+   --> TIME: 2025-06-02 13:46:17 -- STEP: 2008/5792 -- GLOBAL_STEP: 7800
+     | > loss_text_ce: 0.040393561124801636  (0.04518387101839232)
+     | > loss_mel_ce: 3.0478756427764893  (3.367659538628096)
+     | > loss: 0.7720673084259033  (0.8532108522625083)
+     | > current_lr: 5e-06 
+     | > step_time: 0.2072  (0.33570773442427937)
+     | > loader_time: 0.0096  (0.011115385597920515)
+
+
+   --> TIME: 2025-06-02 13:46:46 -- STEP: 2058/5792 -- GLOBAL_STEP: 7850
+     | > loss_text_ce: 0.04632216691970825  (0.04517081452024949)
+     | > loss_mel_ce: 3.593982458114624  (3.366837971527908)
+     | > loss: 0.9100761413574219  (0.8530021963534248)
+     | > current_lr: 5e-06 
+     | > step_time: 0.6081  (0.3358027338402379)
+     | > loader_time: 0.0109  (0.011119001460839988)
+
+
+   --> TIME: 2025-06-02 13:47:17 -- STEP: 2108/5792 -- GLOBAL_STEP: 7900
+     | > loss_text_ce: 0.046658266335725784  (0.04515279076756657)
+     | > loss_mel_ce: 3.4903433322906494  (3.3665078355872438)
+     | > loss: 0.8842504024505615  (0.8529151564983534)
+     | > current_lr: 5e-06 
+     | > step_time: 0.4083  (0.3364675910242141)
+     | > loader_time: 0.0108  (0.011129756353837944)
+
+
+   --> TIME: 2025-06-02 13:47:46 -- STEP: 2158/5792 -- GLOBAL_STEP: 7950
+     | > loss_text_ce: 0.04326501116156578  (0.04511675107844374)
+     | > loss_mel_ce: 3.7370195388793945  (3.365736954072983)
+     | > loss: 0.9450711607933044  (0.8527134262211792)
+     | > current_lr: 5e-06 
+     | > step_time: 0.3271  (0.3362241120554982)
+     | > loader_time: 0.0134  (0.011123432065735718)
+
+
+   --> TIME: 2025-06-02 13:48:15 -- STEP: 2208/5792 -- GLOBAL_STEP: 8000
+     | > loss_text_ce: 0.04933100938796997  (0.04507167210238918)
+     | > loss_mel_ce: 3.107172966003418  (3.364644525707631)
+     | > loss: 0.7891259789466858  (0.8524290493594995)
+     | > current_lr: 5e-06 
+     | > step_time: 0.2153  (0.33627322402553284)
+     | > loader_time: 0.0104  (0.011131968619166944)
+
+
+   --> TIME: 2025-06-02 13:48:43 -- STEP: 2258/5792 -- GLOBAL_STEP: 8050
+     | > loss_text_ce: 0.0509863905608654  (0.04505748897576433)
+     | > loss_mel_ce: 3.511996269226074  (3.3630264050354457)
+     | > loss: 0.8907456398010254  (0.8520209733578247)
+     | > current_lr: 5e-06 
+     | > step_time: 0.4846  (0.3359105811908882)
+     | > loader_time: 0.0105  (0.011143911930182215)
+
+
+   --> TIME: 2025-06-02 13:49:15 -- STEP: 2308/5792 -- GLOBAL_STEP: 8100
+     | > loss_text_ce: 0.04393137991428375  (0.045034569963341514)
+     | > loss_mel_ce: 2.9540281295776367  (3.362665162962478)
+     | > loss: 0.7494899034500122  (0.8519249330411955)
+     | > current_lr: 5e-06 
+     | > step_time: 0.2141  (0.33653108003441157)
+     | > loader_time: 0.0099  (0.011160700325329545)
+
+
+   --> TIME: 2025-06-02 13:49:45 -- STEP: 2358/5792 -- GLOBAL_STEP: 8150
+     | > loss_text_ce: 0.04599863290786743  (0.04497464137914904)
+     | > loss_mel_ce: 3.3983516693115234  (3.363921059098053)
+     | > loss: 0.8610875606536865  (0.8522239249070076)
+     | > current_lr: 5e-06 
+     | > step_time: 0.3547  (0.3368334209039317)
+     | > loader_time: 0.0101  (0.011168763332188992)
+
+
+   --> TIME: 2025-06-02 13:50:13 -- STEP: 2408/5792 -- GLOBAL_STEP: 8200
+     | > loss_text_ce: 0.0410335548222065  (0.04494161312808699)
+     | > loss_mel_ce: 2.8338351249694824  (3.362981556558925)
+     | > loss: 0.7187171578407288  (0.8519807922285653)
+     | > current_lr: 5e-06 
+     | > step_time: 0.1891  (0.33666275852542393)
+     | > loader_time: 0.0106  (0.011154194111839877)
+
+
+   --> TIME: 2025-06-02 13:50:44 -- STEP: 2458/5792 -- GLOBAL_STEP: 8250
+     | > loss_text_ce: 0.04467436298727989  (0.04491210582857818)
+     | > loss_mel_ce: 3.4023947715759277  (3.362881237942576)
+     | > loss: 0.8617672920227051  (0.8519483357546673)
+     | > current_lr: 5e-06 
+     | > step_time: 0.2659  (0.3370825775858832)
+     | > loader_time: 0.0133  (0.011155853725430246)
+
+
+   --> TIME: 2025-06-02 13:51:14 -- STEP: 2508/5792 -- GLOBAL_STEP: 8300
+     | > loss_text_ce: 0.0431218296289444  (0.04488258450210355)
+     | > loss_mel_ce: 3.392765998840332  (3.3614071411759463)
+     | > loss: 0.8589719533920288  (0.8515724312793311)
+     | > current_lr: 5e-06 
+     | > step_time: 0.3281  (0.33738990387087664)
+     | > loader_time: 0.0105  (0.011148617028049297)
+
+
+   --> TIME: 2025-06-02 13:51:43 -- STEP: 2558/5792 -- GLOBAL_STEP: 8350
+     | > loss_text_ce: 0.042658887803554535  (0.0448320767120622)
+     | > loss_mel_ce: 3.3791749477386475  (3.3615849678920893)
+     | > loss: 0.8554584383964539  (0.8516042609993973)
+     | > current_lr: 5e-06 
+     | > step_time: 0.2925  (0.3374677144334692)
+     | > loader_time: 0.0098  (0.011138459217557677)
+
+
+   --> TIME: 2025-06-02 13:52:12 -- STEP: 2608/5792 -- GLOBAL_STEP: 8400
+     | > loss_text_ce: 0.03552236780524254  (0.04479173863623748)
+     | > loss_mel_ce: 2.71285343170166  (3.360946973682912)
+     | > loss: 0.68709397315979  (0.851434677976406)
+     | > current_lr: 5e-06 
+     | > step_time: 0.1824  (0.33746330266349933)
+     | > loader_time: 0.012  (0.01112092354911966)
+
+
+   --> TIME: 2025-06-02 13:52:40 -- STEP: 2658/5792 -- GLOBAL_STEP: 8450
+     | > loss_text_ce: 0.045843303203582764  (0.0447824644737725)
+     | > loss_mel_ce: 3.9877352714538574  (3.3594930995398657)
+     | > loss: 1.0083945989608765  (0.8510688909289527)
+     | > current_lr: 5e-06 
+     | > step_time: 0.2632  (0.337073947003769)
+     | > loader_time: 0.0093  (0.011125095810900938)
+
+
+   --> TIME: 2025-06-02 13:53:08 -- STEP: 2708/5792 -- GLOBAL_STEP: 8500
+     | > loss_text_ce: 0.04365628957748413  (0.04472194665195517)
+     | > loss_mel_ce: 3.502896308898926  (3.358532962189819)
+     | > loss: 0.8866381645202637  (0.8508137271710651)
+     | > current_lr: 5e-06 
+     | > step_time: 0.3144  (0.3368678473299331)
+     | > loader_time: 0.014  (0.011116211136123433)
+
+
+   --> TIME: 2025-06-02 13:53:36 -- STEP: 2758/5792 -- GLOBAL_STEP: 8550
+     | > loss_text_ce: 0.044692471623420715  (0.044700241800292285)
+     | > loss_mel_ce: 3.10170578956604  (3.356952504717499)
+     | > loss: 0.7865995764732361  (0.8504131865769383)
+     | > current_lr: 5e-06 
+     | > step_time: 0.1751  (0.33657982109075396)
+     | > loader_time: 0.0096  (0.011113227234965547)
+
+
+   --> TIME: 2025-06-02 13:54:05 -- STEP: 2808/5792 -- GLOBAL_STEP: 8600
+     | > loss_text_ce: 0.04175104200839996  (0.044659487993995044)
+     | > loss_mel_ce: 3.7599270343780518  (3.3555039678543714)
+     | > loss: 0.950419545173645  (0.8500408639340655)
+     | > current_lr: 5e-06 
+     | > step_time: 0.3366  (0.3363867896234893)
+     | > loader_time: 0.0102  (0.011102559050263819)
+
+
+   --> TIME: 2025-06-02 13:54:33 -- STEP: 2858/5792 -- GLOBAL_STEP: 8650
+     | > loss_text_ce: 0.04298544302582741  (0.04463345923137111)
+     | > loss_mel_ce: 3.1992127895355225  (3.3554093194558448)
+     | > loss: 0.8105495572090149  (0.8500106946582804)
+     | > current_lr: 5e-06 
+     | > step_time: 0.293  (0.3361897283371222)
+     | > loader_time: 0.0101  (0.011095891055566914)
+
+
+   --> TIME: 2025-06-02 13:55:02 -- STEP: 2908/5792 -- GLOBAL_STEP: 8700
+     | > loss_text_ce: 0.04670070484280586  (0.044607090496649)
+     | > loss_mel_ce: 3.638002872467041  (3.354981057827541)
+     | > loss: 0.9211758971214294  (0.8498970369771223)
+     | > current_lr: 5e-06 
+     | > step_time: 0.2085  (0.33621489690127415)
+     | > loader_time: 0.0093  (0.011086543947663218)
+
+
+   --> TIME: 2025-06-02 13:55:31 -- STEP: 2958/5792 -- GLOBAL_STEP: 8750
+     | > loss_text_ce: 0.04170145094394684  (0.044566551267476086)
+     | > loss_mel_ce: 3.448326826095581  (3.35389118762013)
+     | > loss: 0.8725070953369141  (0.849614434604148)
+     | > current_lr: 5e-06 
+     | > step_time: 0.4418  (0.3360969547970708)
+     | > loader_time: 0.0106  (0.01107192184572044)
+
+
+   --> TIME: 2025-06-02 13:55:59 -- STEP: 3008/5792 -- GLOBAL_STEP: 8800
+     | > loss_text_ce: 0.04650944098830223  (0.04452610574786532)
+     | > loss_mel_ce: 3.232806921005249  (3.3524444404434646)
+     | > loss: 0.8198291063308716  (0.8492426364781689)
+     | > current_lr: 5e-06 
+     | > step_time: 0.4633  (0.33591527674109456)
+     | > loader_time: 0.0122  (0.011064917958797276)
+
+
+   --> TIME: 2025-06-02 13:56:28 -- STEP: 3058/5792 -- GLOBAL_STEP: 8850
+     | > loss_text_ce: 0.04007389768958092  (0.04448877311780065)
+     | > loss_mel_ce: 3.012786626815796  (3.3516998684741526)
+     | > loss: 0.7632151246070862  (0.8490471603669237)
+     | > current_lr: 5e-06 
+     | > step_time: 0.2232  (0.33596821498683804)
+     | > loader_time: 0.0119  (0.01105607171866376)
+
+
+   --> TIME: 2025-06-02 13:56:56 -- STEP: 3108/5792 -- GLOBAL_STEP: 8900
+     | > loss_text_ce: 0.03358534350991249  (0.04445769784173796)
+     | > loss_mel_ce: 3.2545511722564697  (3.3518611339221738)
+     | > loss: 0.8220341205596924  (0.8490797078555766)
+     | > current_lr: 5e-06 
+     | > step_time: 0.2089  (0.33573530938481194)
+     | > loader_time: 0.0094  (0.011041443857293623)
+
+
+   --> TIME: 2025-06-02 13:57:26 -- STEP: 3158/5792 -- GLOBAL_STEP: 8950
+     | > loss_text_ce: 0.037105049937963486  (0.04442861938803087)
+     | > loss_mel_ce: 3.5840868949890137  (3.351841525762725)
+     | > loss: 0.9052979946136475  (0.8490675361832911)
+     | > current_lr: 5e-06 
+     | > step_time: 0.4385  (0.33593913925985125)
+     | > loader_time: 0.0127  (0.011041003590222006)
+
+
+   --> TIME: 2025-06-02 13:57:54 -- STEP: 3208/5792 -- GLOBAL_STEP: 9000
+     | > loss_text_ce: 0.04580637440085411  (0.04439754502504062)
+     | > loss_mel_ce: 3.0522005558013916  (3.351788088567833)
+     | > loss: 0.7745017409324646  (0.8490464082754161)
+     | > current_lr: 5e-06 
+     | > step_time: 0.1902  (0.33578265449055483)
+     | > loader_time: 0.0086  (0.011037279245562075)
+
+
+   --> TIME: 2025-06-02 13:58:22 -- STEP: 3258/5792 -- GLOBAL_STEP: 9050
+     | > loss_text_ce: 0.03635700047016144  (0.044366618606934825)
+     | > loss_mel_ce: 3.045945167541504  (3.3514977041529024)
+     | > loss: 0.7705755233764648  (0.8489660805467447)
+     | > current_lr: 5e-06 
+     | > step_time: 0.2663  (0.3356093359256983)
+     | > loader_time: 0.0145  (0.011036152924548766)
+
+
+   --> TIME: 2025-06-02 13:58:52 -- STEP: 3308/5792 -- GLOBAL_STEP: 9100
+     | > loss_text_ce: 0.0383525975048542  (0.04434114113729335)
+     | > loss_mel_ce: 3.439059257507324  (3.3502290691401098)
+     | > loss: 0.8693529367446899  (0.848642552417884)
+     | > current_lr: 5e-06 
+     | > step_time: 0.2088  (0.33574586695886793)
+     | > loader_time: 0.0124  (0.011047434129334404)
+
+
+   --> TIME: 2025-06-02 13:59:20 -- STEP: 3358/5792 -- GLOBAL_STEP: 9150
+     | > loss_text_ce: 0.0411563478410244  (0.04431965041788048)
+     | > loss_mel_ce: 3.5818145275115967  (3.3501473820017798)
+     | > loss: 0.9057427048683167  (0.8486167579587544)
+     | > current_lr: 5e-06 
+     | > step_time: 0.333  (0.3356304534352067)
+     | > loader_time: 0.0102  (0.01104049429856572)
+
+
+   --> TIME: 2025-06-02 13:59:49 -- STEP: 3408/5792 -- GLOBAL_STEP: 9200
+     | > loss_text_ce: 0.04104316234588623  (0.044287355214427344)
+     | > loss_mel_ce: 3.3351967334747314  (3.349034378436249)
+     | > loss: 0.844059944152832  (0.8483304332224694)
+     | > current_lr: 5e-06 
+     | > step_time: 0.3537  (0.3354787281561352)
+     | > loader_time: 0.0143  (0.01104401031010586)
+
+
+   --> TIME: 2025-06-02 14:00:17 -- STEP: 3458/5792 -- GLOBAL_STEP: 9250
+     | > loss_text_ce: 0.04712622985243797  (0.04426270988896412)
+     | > loss_mel_ce: 3.5113916397094727  (3.3474128444874607)
+     | > loss: 0.8896294832229614  (0.8479188883352856)
+     | > current_lr: 5e-06 
+     | > step_time: 0.4075  (0.3352865505383985)
+     | > loader_time: 0.0109  (0.011042501417593558)
+
+
+   --> TIME: 2025-06-02 14:00:47 -- STEP: 3508/5792 -- GLOBAL_STEP: 9300
+     | > loss_text_ce: 0.0433344691991806  (0.04423293619052409)
+     | > loss_mel_ce: 3.186034679412842  (3.3471053151611314)
+     | > loss: 0.8073422908782959  (0.8478345625960566)
+     | > current_lr: 5e-06 
+     | > step_time: 0.3399  (0.33543082470225155)
+     | > loader_time: 0.0136  (0.011043719921851497)
+
+
+   --> TIME: 2025-06-02 14:01:15 -- STEP: 3558/5792 -- GLOBAL_STEP: 9350
+     | > loss_text_ce: 0.0385572724044323  (0.04420625756611444)
+     | > loss_mel_ce: 3.43131160736084  (3.3458053263604737)
+     | > loss: 0.8674672245979309  (0.8475028957591555)
+     | > current_lr: 5e-06 
+     | > step_time: 0.2685  (0.3353362489510554)
+     | > loader_time: 0.0128  (0.011042438242259877)
+
+
+   --> TIME: 2025-06-02 14:01:45 -- STEP: 3608/5792 -- GLOBAL_STEP: 9400
+     | > loss_text_ce: 0.04384002462029457  (0.04419183030212402)
+     | > loss_mel_ce: 3.5863473415374756  (3.3455215218722683)
+     | > loss: 0.9075468182563782  (0.8474283378463888)
+     | > current_lr: 5e-06 
+     | > step_time: 0.3275  (0.33544651070349557)
+     | > loader_time: 0.01  (0.011052065572294573)
+
+
+   --> TIME: 2025-06-02 14:02:12 -- STEP: 3658/5792 -- GLOBAL_STEP: 9450
+     | > loss_text_ce: 0.04530433937907219  (0.04417096679045439)
+     | > loss_mel_ce: 3.4153072834014893  (3.3448606153991842)
+     | > loss: 0.865152895450592  (0.8472578953595157)
+     | > current_lr: 5e-06 
+     | > step_time: 0.314  (0.3351731634061808)
+     | > loader_time: 0.0125  (0.011047935929045499)
+
+
+   --> TIME: 2025-06-02 14:02:42 -- STEP: 3708/5792 -- GLOBAL_STEP: 9500
+     | > loss_text_ce: 0.0346110537648201  (0.04413479716469813)
+     | > loss_mel_ce: 3.222661018371582  (3.3444809126622466)
+     | > loss: 0.8143180012702942  (0.8471539272588174)
+     | > current_lr: 5e-06 
+     | > step_time: 0.3003  (0.33524002194276065)
+     | > loader_time: 0.011  (0.011049625685814244)
+
+
+   --> TIME: 2025-06-02 14:03:09 -- STEP: 3758/5792 -- GLOBAL_STEP: 9550
+     | > loss_text_ce: 0.04709426686167717  (0.04409706911509297)
+     | > loss_mel_ce: 3.4125912189483643  (3.343986088002597)
+     | > loss: 0.8649213910102844  (0.8470207890213136)
+     | > current_lr: 5e-06 
+     | > step_time: 0.4286  (0.33501876009595527)
+     | > loader_time: 0.0134  (0.011049652048855007)
+
+
+   --> TIME: 2025-06-02 14:03:38 -- STEP: 3808/5792 -- GLOBAL_STEP: 9600
+     | > loss_text_ce: 0.041285887360572815  (0.044069222885305306)
+     | > loss_mel_ce: 3.439079523086548  (3.3440659681037688)
+     | > loss: 0.8700913786888123  (0.8470337975676318)
+     | > current_lr: 5e-06 
+     | > step_time: 0.2723  (0.334857460267905)
+     | > loader_time: 0.0094  (0.011047247202456485)
+
+
+   --> TIME: 2025-06-02 14:04:07 -- STEP: 3858/5792 -- GLOBAL_STEP: 9650
+     | > loss_text_ce: 0.0423085018992424  (0.04404395498178935)
+     | > loss_mel_ce: 3.5037617683410645  (3.343483012102007)
+     | > loss: 0.8865175843238831  (0.8468817415912264)
+     | > current_lr: 5e-06 
+     | > step_time: 0.41  (0.33492562775787443)
+     | > loader_time: 0.0131  (0.011050497329933387)
+
+
+   --> TIME: 2025-06-02 14:04:37 -- STEP: 3908/5792 -- GLOBAL_STEP: 9700
+     | > loss_text_ce: 0.04542999714612961  (0.04403033828412392)
+     | > loss_mel_ce: 2.891977071762085  (3.342988579214656)
+     | > loss: 0.7343517541885376  (0.8467547291987015)
+     | > current_lr: 5e-06 
+     | > step_time: 0.1907  (0.335096147006708)
+     | > loader_time: 0.0098  (0.011049572435591811)
+
+
+   --> TIME: 2025-06-02 14:05:07 -- STEP: 3958/5792 -- GLOBAL_STEP: 9750
+     | > loss_text_ce: 0.0466097928583622  (0.044010729583553475)
+     | > loss_mel_ce: 3.508066177368164  (3.3432548332238445)
+     | > loss: 0.8886690139770508  (0.8468163905419621)
+     | > current_lr: 5e-06 
+     | > step_time: 0.3011  (0.3353043986788423)
+     | > loader_time: 0.013  (0.011049168347228097)
+
+
+   --> TIME: 2025-06-02 14:05:38 -- STEP: 4008/5792 -- GLOBAL_STEP: 9800
+     | > loss_text_ce: 0.04837203770875931  (0.04398115760806629)
+     | > loss_mel_ce: 3.3902106285095215  (3.3430067203834857)
+     | > loss: 0.859645664691925  (0.8467469693955307)
+     | > current_lr: 5e-06 
+     | > step_time: 0.313  (0.33554493548151554)
+     | > loader_time: 0.01  (0.011050231026080303)
+
+
+   --> TIME: 2025-06-02 14:06:08 -- STEP: 4058/5792 -- GLOBAL_STEP: 9850
+     | > loss_text_ce: 0.04397377371788025  (0.043955045121553474)
+     | > loss_mel_ce: 3.5290706157684326  (3.342233718645989)
+     | > loss: 0.8932610750198364  (0.8465471908251834)
+     | > current_lr: 5e-06 
+     | > step_time: 0.4423  (0.3356970511615546)
+     | > loader_time: 0.0139  (0.01104889202728947)
+
+
+   --> TIME: 2025-06-02 14:06:36 -- STEP: 4108/5792 -- GLOBAL_STEP: 9900
+     | > loss_text_ce: 0.034252800047397614  (0.04393632656793966)
+     | > loss_mel_ce: 3.536194086074829  (3.3410506406916842)
+     | > loss: 0.8926117420196533  (0.846246741690329)
+     | > current_lr: 5e-06 
+     | > step_time: 0.2459  (0.33544389101175215)
+     | > loader_time: 0.0095  (0.011051703973545407)
+
+
+   --> TIME: 2025-06-02 14:07:05 -- STEP: 4158/5792 -- GLOBAL_STEP: 9950
+     | > loss_text_ce: 0.03575450927019119  (0.04390224255218264)
+     | > loss_mel_ce: 2.652836799621582  (3.3396065647983955)
+     | > loss: 0.672147810459137  (0.8458772017268847)
+     | > current_lr: 5e-06 
+     | > step_time: 0.3049  (0.33547147149475504)
+     | > loader_time: 0.0129  (0.01104814040655123)
+
+
+   --> TIME: 2025-06-02 14:07:34 -- STEP: 4208/5792 -- GLOBAL_STEP: 10000
+     | > loss_text_ce: 0.044553492218256  (0.04387786551847873)
+     | > loss_mel_ce: 3.6881346702575684  (3.3393874958780767)
+     | > loss: 0.9331720471382141  (0.8458163402549665)
+     | > current_lr: 5e-06 
+     | > step_time: 0.24  (0.33554055975190633)
+     | > loader_time: 0.0136  (0.011045128107070909)
+
+
+   --> TIME: 2025-06-02 14:08:05 -- STEP: 4258/5792 -- GLOBAL_STEP: 10050
+     | > loss_text_ce: 0.04713189974427223  (0.043857144453016465)
+     | > loss_mel_ce: 3.250263214111328  (3.3402409040631116)
+     | > loss: 0.8243488073348999  (0.8460245119847841)
+     | > current_lr: 5e-06 
+     | > step_time: 0.4278  (0.33582793624818846)
+     | > loader_time: 0.0109  (0.011045684160453815)
+
+
+   --> TIME: 2025-06-02 14:08:36 -- STEP: 4308/5792 -- GLOBAL_STEP: 10100
+     | > loss_text_ce: 0.04142797365784645  (0.04383900594753162)
+     | > loss_mel_ce: 3.029611110687256  (3.339805153861793)
+     | > loss: 0.7677597999572754  (0.8459110398229447)
+     | > current_lr: 5e-06 
+     | > step_time: 0.2426  (0.3360530724653853)
+     | > loader_time: 0.0125  (0.01104891161843372)
+
+
+   --> TIME: 2025-06-02 14:09:06 -- STEP: 4358/5792 -- GLOBAL_STEP: 10150
+     | > loss_text_ce: 0.04575802758336067  (0.04381198164147425)
+     | > loss_mel_ce: 3.206411838531494  (3.3394206824681243)
+     | > loss: 0.8130424618721008  (0.8458081659180897)
+     | > current_lr: 5e-06 
+     | > step_time: 0.4958  (0.33611179339114383)
+     | > loader_time: 0.0112  (0.011051389699465848)
+
+
+   --> TIME: 2025-06-02 14:09:35 -- STEP: 4408/5792 -- GLOBAL_STEP: 10200
+     | > loss_text_ce: 0.04181518405675888  (0.04377960568140123)
+     | > loss_mel_ce: 2.947584629058838  (3.3390845118763206)
+     | > loss: 0.7473499774932861  (0.8457160292703737)
+     | > current_lr: 5e-06 
+     | > step_time: 0.2769  (0.33606701202271466)
+     | > loader_time: 0.0094  (0.011053307681680795)
+
+
+   --> TIME: 2025-06-02 14:10:06 -- STEP: 4458/5792 -- GLOBAL_STEP: 10250
+     | > loss_text_ce: 0.04814762994647026  (0.0437453800015162)
+     | > loss_mel_ce: 2.8046014308929443  (3.3391340181077727)
+     | > loss: 0.7131872773170471  (0.8457198493841137)
+     | > current_lr: 5e-06 
+     | > step_time: 0.2371  (0.3363838041240513)
+     | > loader_time: 0.0094  (0.011047967242895073)
+
+
+   --> TIME: 2025-06-02 14:10:36 -- STEP: 4508/5792 -- GLOBAL_STEP: 10300
+     | > loss_text_ce: 0.045141834765672684  (0.043721168837076246)
+     | > loss_mel_ce: 3.2513136863708496  (3.3381951364036566)
+     | > loss: 0.8241139054298401  (0.8454790762043034)
+     | > current_lr: 5e-06 
+     | > step_time: 0.2119  (0.3364703716063989)
+     | > loader_time: 0.0097  (0.011043780920456853)
+
+
+   --> TIME: 2025-06-02 14:11:04 -- STEP: 4558/5792 -- GLOBAL_STEP: 10350
+     | > loss_text_ce: 0.046194713562726974  (0.04369950713651088)
+     | > loss_mel_ce: 3.618640184402466  (3.3370097836364305)
+     | > loss: 0.9162087440490723  (0.8451773226261132)
+     | > current_lr: 5e-06 
+     | > step_time: 0.4156  (0.3364097547405593)
+     | > loader_time: 0.013  (0.011043213553049927)
+
+
+   --> TIME: 2025-06-02 14:11:34 -- STEP: 4608/5792 -- GLOBAL_STEP: 10400
+     | > loss_text_ce: 0.044357966631650925  (0.0436816014767424)
+     | > loss_mel_ce: 3.656270742416382  (3.336337390355765)
+     | > loss: 0.9251571893692017  (0.8450047479062854)
+     | > current_lr: 5e-06 
+     | > step_time: 0.2739  (0.3365074931126504)
+     | > loader_time: 0.0125  (0.011044854091273402)
+
+
+   --> TIME: 2025-06-02 14:12:03 -- STEP: 4658/5792 -- GLOBAL_STEP: 10450
+     | > loss_text_ce: 0.03670710697770119  (0.04365839876879141)
+     | > loss_mel_ce: 2.583122968673706  (3.335039994193535)
+     | > loss: 0.6549575328826904  (0.8446745981777001)
+     | > current_lr: 5e-06 
+     | > step_time: 0.1691  (0.336374844487208)
+     | > loader_time: 0.0109  (0.011051791632518373)
+
+
+   --> TIME: 2025-06-02 14:12:31 -- STEP: 4708/5792 -- GLOBAL_STEP: 10500
+     | > loss_text_ce: 0.04842893406748772  (0.04363689536684546)
+     | > loss_mel_ce: 3.4084410667419434  (3.33495435723837)
+     | > loss: 0.8642175197601318  (0.8446478130788037)
+     | > current_lr: 5e-06 
+     | > step_time: 0.1822  (0.33627098611379563)
+     | > loader_time: 0.0117  (0.011046174631953332)
+
+
+   --> TIME: 2025-06-02 14:13:01 -- STEP: 4758/5792 -- GLOBAL_STEP: 10550
+     | > loss_text_ce: 0.042808979749679565  (0.043614106523299764)
+     | > loss_mel_ce: 3.381239652633667  (3.3350225664477926)
+     | > loss: 0.8560121655464172  (0.8446591681745581)
+     | > current_lr: 5e-06 
+     | > step_time: 0.293  (0.3363845983740162)
+     | > loader_time: 0.0131  (0.011052533276394754)
+
+
+   --> TIME: 2025-06-02 14:13:31 -- STEP: 4808/5792 -- GLOBAL_STEP: 10600
+     | > loss_text_ce: 0.04915136843919754  (0.04358621223631336)
+     | > loss_mel_ce: 2.8472352027893066  (3.3347837737515045)
+     | > loss: 0.7240966558456421  (0.8445924964180201)
+     | > current_lr: 5e-06 
+     | > step_time: 0.1576  (0.3364169633329015)
+     | > loader_time: 0.0093  (0.011056910884162139)
+
+
+   --> TIME: 2025-06-02 14:14:02 -- STEP: 4858/5792 -- GLOBAL_STEP: 10650
+     | > loss_text_ce: 0.038126781582832336  (0.0435614236114802)
+     | > loss_mel_ce: 3.1234774589538574  (3.3353497294939896)
+     | > loss: 0.790401041507721  (0.8447277882109941)
+     | > current_lr: 5e-06 
+     | > step_time: 0.2953  (0.3366490865548782)
+     | > loader_time: 0.0091  (0.01105543447159208)
+
+
+   --> TIME: 2025-06-02 14:14:32 -- STEP: 4908/5792 -- GLOBAL_STEP: 10700
+     | > loss_text_ce: 0.04095133766531944  (0.043551660765238684)
+     | > loss_mel_ce: 3.398160457611084  (3.335550274903345)
+     | > loss: 0.8597779273986816  (0.8447754838623056)
+     | > current_lr: 5e-06 
+     | > step_time: 0.2428  (0.33676366579658956)
+     | > loader_time: 0.0128  (0.011054953036506476)
+
+
+   --> TIME: 2025-06-02 14:15:02 -- STEP: 4958/5792 -- GLOBAL_STEP: 10750
+     | > loss_text_ce: 0.04319563880562782  (0.0435207473597263)
+     | > loss_mel_ce: 2.871285915374756  (3.335179869231316)
+     | > loss: 0.7286204099655151  (0.844675154106998)
+     | > current_lr: 5e-06 
+     | > step_time: 0.1926  (0.33679866516671897)
+     | > loader_time: 0.0087  (0.011052699087126775)
+
+
+   --> TIME: 2025-06-02 14:15:30 -- STEP: 5008/5792 -- GLOBAL_STEP: 10800
+     | > loss_text_ce: 0.04007870703935623  (0.043486832795789604)
+     | > loss_mel_ce: 3.0385284423828125  (3.3343036392340646)
+     | > loss: 0.7696517705917358  (0.8444476179516727)
+     | > current_lr: 5e-06 
+     | > step_time: 0.3957  (0.3367528500267494)
+     | > loader_time: 0.011  (0.011056139684332816)
+
+
+   --> TIME: 2025-06-02 14:16:00 -- STEP: 5058/5792 -- GLOBAL_STEP: 10850
+     | > loss_text_ce: 0.04264426231384277  (0.04346949416229308)
+     | > loss_mel_ce: 3.440608263015747  (3.333787598596726)
+     | > loss: 0.8708131313323975  (0.8443142731536648)
+     | > current_lr: 5e-06 
+     | > step_time: 0.3941  (0.3367869499032324)
+     | > loader_time: 0.0124  (0.01105723639352842)
+
+
+   --> TIME: 2025-06-02 14:16:28 -- STEP: 5108/5792 -- GLOBAL_STEP: 10900
+     | > loss_text_ce: 0.034066058695316315  (0.043442419822748615)
+     | > loss_mel_ce: 3.139413595199585  (3.3330060827050625)
+     | > loss: 0.7933698892593384  (0.8441121256388804)
+     | > current_lr: 5e-06 
+     | > step_time: 0.1858  (0.33673182608284047)
+     | > loader_time: 0.0091  (0.011057687393836698)
+
+
+   --> TIME: 2025-06-02 14:16:56 -- STEP: 5158/5792 -- GLOBAL_STEP: 10950
+     | > loss_text_ce: 0.04518938437104225  (0.04340843048297699)
+     | > loss_mel_ce: 3.1620471477508545  (3.3323658000780383)
+     | > loss: 0.8018091320991516  (0.843943557682865)
+     | > current_lr: 5e-06 
+     | > step_time: 0.2567  (0.3365875314307241)
+     | > loader_time: 0.0094  (0.011058349256230763)
+
+
+   --> TIME: 2025-06-02 14:17:24 -- STEP: 5208/5792 -- GLOBAL_STEP: 11000
+     | > loss_text_ce: 0.04442836344242096  (0.04338480853733139)
+     | > loss_mel_ce: 3.0665812492370605  (3.3311940293799163)
+     | > loss: 0.7777523994445801  (0.8436447095280412)
+     | > current_lr: 5e-06 
+     | > step_time: 0.2419  (0.3364083989882436)
+     | > loader_time: 0.0102  (0.011055767169928952)
+
+
+   --> TIME: 2025-06-02 14:17:53 -- STEP: 5258/5792 -- GLOBAL_STEP: 11050
+     | > loss_text_ce: 0.04343422129750252  (0.04335219707671583)
+     | > loss_mel_ce: 2.8784594535827637  (3.3301943961365135)
+     | > loss: 0.7304733991622925  (0.8433866483510418)
+     | > current_lr: 5e-06 
+     | > step_time: 0.3317  (0.3363045960517387)
+     | > loader_time: 0.0101  (0.011054212260763195)
+
+
+   --> TIME: 2025-06-02 14:18:21 -- STEP: 5308/5792 -- GLOBAL_STEP: 11100
+     | > loss_text_ce: 0.04893643409013748  (0.04332292118490856)
+     | > loss_mel_ce: 3.3955016136169434  (3.3295369724986603)
+     | > loss: 0.8611094951629639  (0.8432149734929163)
+     | > current_lr: 5e-06 
+     | > step_time: 0.326  (0.33622330490053426)
+     | > loader_time: 0.0105  (0.011051318837721364)
+
+
+   --> TIME: 2025-06-02 14:18:52 -- STEP: 5358/5792 -- GLOBAL_STEP: 11150
+     | > loss_text_ce: 0.044002778828144073  (0.04330665982714582)
+     | > loss_mel_ce: 4.1681904792785645  (3.3292624402642472)
+     | > loss: 1.0530483722686768  (0.8431422751134073)
+     | > current_lr: 5e-06 
+     | > step_time: 0.5552  (0.3363442467237067)
+     | > loader_time: 0.0151  (0.011057644159146863)
+
+
+   --> TIME: 2025-06-02 14:19:20 -- STEP: 5408/5792 -- GLOBAL_STEP: 11200
+     | > loss_text_ce: 0.038751423358917236  (0.043278468308141)
+     | > loss_mel_ce: 3.3097198009490967  (3.3285519173864784)
+     | > loss: 0.8371177911758423  (0.8429575964764371)
+     | > current_lr: 5e-06 
+     | > step_time: 0.2749  (0.3362092402647945)
+     | > loader_time: 0.0135  (0.011058031013731425)
+
+
+   --> TIME: 2025-06-02 14:19:48 -- STEP: 5458/5792 -- GLOBAL_STEP: 11250
+     | > loss_text_ce: 0.041902702301740646  (0.04325453944254458)
+     | > loss_mel_ce: 2.8589980602264404  (3.3282593212620424)
+     | > loss: 0.7252252101898193  (0.8428784652632549)
+     | > current_lr: 5e-06 
+     | > step_time: 0.2551  (0.33609242695248986)
+     | > loader_time: 0.0099  (0.011059924594296262)
+
+
+   --> TIME: 2025-06-02 14:20:16 -- STEP: 5508/5792 -- GLOBAL_STEP: 11300
+     | > loss_text_ce: 0.036019548773765564  (0.0432351567869034)
+     | > loss_mel_ce: 2.981379270553589  (3.3276002054294125)
+     | > loss: 0.7543497085571289  (0.842708840619937)
+     | > current_lr: 5e-06 
+     | > step_time: 0.1859  (0.3359816792483949)
+     | > loader_time: 0.0119  (0.011056781118125613)
+
+
+   --> TIME: 2025-06-02 14:20:47 -- STEP: 5558/5792 -- GLOBAL_STEP: 11350
+     | > loss_text_ce: 0.04409126192331314  (0.04321853997813861)
+     | > loss_mel_ce: 3.3099846839904785  (3.3272555071314964)
+     | > loss: 0.8385189771652222  (0.842618511869736)
+     | > current_lr: 5e-06 
+     | > step_time: 0.3187  (0.33623436845091076)
+     | > loader_time: 0.0121  (0.011060114171102605)
+
+
+   --> TIME: 2025-06-02 14:21:16 -- STEP: 5608/5792 -- GLOBAL_STEP: 11400
+     | > loss_text_ce: 0.03766907751560211  (0.043189963759633916)
+     | > loss_mel_ce: 3.2820019721984863  (3.325929513628915)
+     | > loss: 0.8299177885055542  (0.8422798694412984)
+     | > current_lr: 5e-06 
+     | > step_time: 0.3434  (0.3362132219037052)
+     | > loader_time: 0.0099  (0.011059219843990964)
+
+
+   --> TIME: 2025-06-02 14:21:45 -- STEP: 5658/5792 -- GLOBAL_STEP: 11450
+     | > loss_text_ce: 0.0405101515352726  (0.04316613374270707)
+     | > loss_mel_ce: 3.357116937637329  (3.325798077269804)
+     | > loss: 0.8494067788124084  (0.8422410528686782)
+     | > current_lr: 5e-06 
+     | > step_time: 0.2667  (0.3361453908035056)
+     | > loader_time: 0.0096  (0.011055887103628554)
+
+
+   --> TIME: 2025-06-02 14:22:11 -- STEP: 5708/5792 -- GLOBAL_STEP: 11500
+     | > loss_text_ce: 0.033331647515296936  (0.04313662911477462)
+     | > loss_mel_ce: 2.8956968784332275  (3.3249534104951377)
+     | > loss: 0.7322571277618408  (0.8420225100117953)
+     | > current_lr: 5e-06 
+     | > step_time: 0.1596  (0.3358227129406929)
+     | > loader_time: 0.0095  (0.011055266556409003)
+
+
+   --> TIME: 2025-06-02 14:22:39 -- STEP: 5758/5792 -- GLOBAL_STEP: 11550
+     | > loss_text_ce: 0.042040519416332245  (0.04311639427965342)
+     | > loss_mel_ce: 2.8961079120635986  (3.3243683234152175)
+     | > loss: 0.7345371246337891  (0.8418711795600674)
+     | > current_lr: 5e-06 
+     | > step_time: 0.2629  (0.33571524755041415)
+     | > loader_time: 0.0125  (0.011058961322382958)
+
+
+ > EVALUATION 
+
+
+  --> EVAL PERFORMANCE
+     | > avg_loader_time: 0.20065663089859956 (-0.0009856695509226354)
+     | > avg_loss_text_ce: 0.0388087550504396 (-0.00597116267992618)
+     | > avg_loss_mel_ce: 3.061407669115872 (-0.1385840406525629)
+     | > avg_loss: 3.100216426418325 (-0.14455520433221114)
+
+ > BEST MODEL : xTTS/checkpoints/GPT_XTTS_FT-June-02-2025_12+22PM-0000000/best_model_11584.pth