{
  "architectures": [
    "SOPImageCls"
  ],
  "attn_patch_size": 6,
  "class_weight_layer": "model.fc4",
  "finetune_layers": [
    "model.fc4"
  ],
  "freeze_projection": false,
  "group_gen_blur_ks1": 1,
  "group_gen_blur_ks2": -1,
  "group_gen_blur_sigma1": 1,
  "group_gen_blur_sigma2": -1,
  "group_gen_scale": 0.06,
  "group_gen_temp_alpha": 1,
  "group_gen_temp_beta": 1,
  "group_sel_scale": 10,
  "hidden_size": 32,
  "image_size": 66,
  "input_hidden_size": 32,
  "json_file": "/shared_data0/weiqiuy/sop/notebooks/exps/cosmogrid_lr5e-06_tgtnnz0.2_gg0.0600_gs10.0000_ft_identify_fixk_scratch_ks1_segpatch_4h/best",
  "num_channels": 1,
  "num_heads": 4,
  "num_labels": 2,
  "num_masks_max": 200,
  "num_masks_sample": 20,
  "projected_input_scale": 1,
  "torch_dtype": "float32",
  "transformers_version": "4.34.0"
}