exp0/config.json

{
  "train_jsonl": "data/groups_train.jsonl",
  "valid_jsonl": "data/groups_valid.jsonl",
  "run_dir": "runs/exp0",
  "backbone": "google/mt5-small",
  "num_latents": 16,
  "latent_dropout": 0.1,
  "latent_noise_std": 0.01,
  "batch_size": 4,
  "grad_accum": 8,
  "lr": 0.0001,
  "epochs": 1,
  "max_doc_len": 256,
  "max_sum_len": 64,
  "lambda_align": 0.1,
  "tau": 0.07,
  "eval_every": 500,
  "max_train_examples": 0,
  "max_valid_examples": 0
}