ttts/gpt/config.json

{
  "train": {
    "train_steps":100000,
    "val_freq" : 100,
    "save_freq" : 1000,
    "keep_ckpts" : 3,
    "lr":1e-4,
    "logs_folder":"ttts/gpt/logs",
    "text_weight":0.01,
    "mel_weight":1,
    "accumulate_num":16
  },
  "dataset": {
    "path":"/home/hyc/tortoise_plus_zh/ttts/datasets/databaker_data.jsonl"
  },
  "gpt":{
    "model_dim":1024,
    "max_mel_tokens":604,
    "max_text_tokens":402,
    "heads":16,
    "use_mel_codes_as_input":true,
    "layers":15,
    "number_text_tokens":256,
    "number_mel_codes":8194,
    "start_text_token":255,
    "train_solo_embeddings":false
  },
  "dataloader":
  {
    "batch_size" : 8,
    "shuffle": true,
    "num_workers" : 64,
    "drop_last":true, 
    "pin_memory":true
  },
  "comment":{
    "sampler":"sampler", 
    "collate_fn":"collate_fn"
  }
}