{ "architectures": [ "HieraForPreTraining" ], "decoder_depth": 1, "decoder_hidden_size": 2, "decoder_num_heads": 1, "depths": [ 1, 1, 1, 1 ], "drop_path_rate": 0.0, "dtype": "float32", "embed_dim": 8, "embed_dim_multiplier": 2.0, "hidden_act": "gelu", "hidden_size": 64, "image_size": [ 64, 64 ], "initializer_range": 0.02, "layer_norm_eps": 1e-06, "layer_norm_init": 1.0, "mask_ratio": 0.6, "masked_unit_attention": [ true, true, false, false ], "masked_unit_size": [ 8, 8 ], "mlp_ratio": 1.0, "model_type": "hiera", "normalize_pixel_loss": true, "num_channels": 3, "num_heads": [ 1, 1, 1, 1 ], "num_query_pool": 3, "out_features": [ "stage4" ], "out_indices": [ 4 ], "patch_padding": [ 3, 3 ], "patch_size": [ 7, 7 ], "patch_stride": [ 4, 4 ], "query_stride": [ 2, 2 ], "stage_names": [ "stem", "stage1", "stage2", "stage3", "stage4" ], "transformers_version": "5.16.0.dev0" }