adv2_smoke/config.json

{
  "train_jsonl": "data/groups_train.jsonl",
  "valid_jsonl": "data/groups_valid.jsonl",
  "run_dir": "runs/adv2_smoke",
  "backbone": "google/mt5-small",
  "num_latents": 16,
  "latent_dropout": 0.1,
  "latent_noise_std": 0.01,
  "batch_size": 4,
  "grad_accum": 8,
  "epochs": 1,
  "max_doc_len": 256,
  "max_sum_len": 64,
  "eval_every": 200,
  "max_train_examples": 2000,
  "max_valid_examples": 200,
  "lambda_align": 0.2,
  "tau": 0.07,
  "lambda_langadv": 1.0,
  "lang_steps": 3,
  "lambda_varcov": 50.0,
  "var_target_std": 0.05,
  "lr_model": 0.0001,
  "lr_lang": 0.01
}