config.json

{
  "d_model": 32,
  "expansion": 32,
  "k": 32,
  "lr": 0.0003,
  "batch_size": 4096,
  "seed": 0,
  "unit_norm_decoder": true,
  "n_features": 1024,
  "layer": 1,
  "base_source": "trained:bilinear-2layer-seed2",
  "replicate": 9,
  "init_seed": 0,
  "train_subset_seed": 1021009
}