{ | |
"exp_name": "F5TTS_Base", | |
"learning_rate": 5e-05, | |
"batch_size_per_gpu": 3200, | |
"batch_size_type": "frame", | |
"max_samples": 64, | |
"grad_accumulation_steps": 1, | |
"max_grad_norm": 1, | |
"epochs": 200, | |
"num_warmup_updates": 2500, | |
"save_per_updates": 7000, | |
"keep_last_n_checkpoints": 0, | |
"last_per_updates": 1500, | |
"finetune": true, | |
"file_checkpoint_train": "", | |
"tokenizer_type": "char", | |
"tokenizer_file": "", | |
"mixed_precision": "bf16", | |
"logger": "wandb", | |
"bnb_optimizer": false | |
} |