configs/small.json

{
  "d_pert": 16,
  "hidden": 64,
  "depth": 2,
  "epochs": 2,
  "batch_size": 64,
  "lr": 0.001,
  "weight_decay": 1e-05,
  "lam_tan": 1.0,
  "lam_semi": 0.5,
  "lam_reg": 0.0001,
  "lam_dist": 0.0,
  "match": "random",
  "seed": 0,
  "device": "cpu",
  "threads": 4
}