File size: 829 Bytes
e7403cc e3069d7 e7403cc |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 |
{
"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"
}
|