checkpoints/multiscale_v2_nsd_block7/run_args.json

{
  "data_path": null,
  "hdf5_path": "/n/holylfs06/LABS/kempner_fellow_binxuwang/Users/binxuwang/Datasets/NSD_human_fMRI/stimuli/nsd_stimuli.hdf5",
  "hdf5_key": "imgBrick",
  "image_root": null,
  "image_column": "image",
  "dino_model": "dinov2_vitb14",
  "dino_repo_path": "/n/holylfs06/LABS/kempner_fellow_binxuwang/Users/binxuwang/torch_cache/hub/facebookresearch_dinov2_main",
  "image_size": 224,
  "dino_block": 7,
  "g0_dist_min": 6,
  "g0_dist_max": 8,
  "g1_pos_dist_min": 3,
  "g1_pos_dist_max": 5,
  "g1_neg_dist_min": 1,
  "g1_neg_dist_max": 2,
  "g2_pos_dist": 1,
  "g2_neg_dist_min": 4,
  "batch_size": 512,
  "batch_size_images": 8,
  "pairs_per_image": 64,
  "num_workers": 4,
  "steps": 100000,
  "dict_size": 16384,
  "k": 32,
  "group_fractions": [
    0.25,
    0.25,
    0.25,
    0.25
  ],
  "group_weights": null,
  "lr": null,
  "auxk_alpha": 0.03125,
  "temp_alpha": 0.1,
  "warmup_steps": 1000,
  "decay_start": null,
  "threshold_beta": 0.999,
  "threshold_start_step": 1000,
  "seed": 0,
  "layer": 0,
  "lm_name": "dinov2_vitb14",
  "wandb_name": "MultiscaleSpatialSAE_v2_NSD75k_Block7",
  "submodule_name": "x_norm_patchtokens",
  "device": "cuda",
  "save_dir": "/n/home12/binxuwang/T-SAE-Spatial/ckpts_multiscale_v2_nsd_block7",
  "save_every": 10000,
  "init_from_ckpt": null,
  "log_every": 200,
  "activation_dim": 768
}