| { | |
| "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" | |
| } | |