| { | |
| "attn_dropout": 0.1, | |
| "architectures": [ | |
| "MARModel" | |
| ], | |
| "model_type": "mar", | |
| "model_name_or_path": "jadechoghari/mar", | |
| "auto_map": { | |
| "AutoConfig": "modeling.MARConfig", | |
| "AutoModel": "modeling.MARModel" | |
| }, | |
| "buffer_size": 64, | |
| "class_num": 1000, | |
| "decoder_depth": 16, | |
| "decoder_embed_dim": 1024, | |
| "decoder_num_heads": 16, | |
| "diffloss_d": 3, | |
| "diffloss_w": 1024, | |
| "diffusion_batch_mul": 4, | |
| "encoder_depth": 16, | |
| "encoder_embed_dim": 1024, | |
| "encoder_num_heads": 16, | |
| "grad_checkpointing": false, | |
| "img_size": 256, | |
| "label_drop_prob": 0.1, | |
| "mask_ratio_min": 0.7, | |
| "mlp_ratio": 4.0, | |
| "model_type": "mar", | |
| "norm_layer": "LayerNorm", | |
| "num_sampling_steps": "100", | |
| "patch_size": 1, | |
| "proj_dropout": 0.1, | |
| "transformers_version": "4.44.2", | |
| "vae_embed_dim": 16, | |
| "vae_stride": 16 | |
| } | |