jadechoghari commited on
Commit
d059bcc
1 Parent(s): 5f26979

Create config.json

Browse files
Files changed (1) hide show
  1. config.json +36 -0
config.json ADDED
@@ -0,0 +1,36 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "attn_dropout": 0.1,
3
+ "architectures": [
4
+ "MARModel"
5
+ ],
6
+ "model_type": "mar",
7
+ "model_name_or_path": "jadechoghari/mar",
8
+ "auto_map": {
9
+ "AutoConfig": "modeling.MARConfig",
10
+ "AutoModel": "modeling.MARModel"
11
+ },
12
+ "buffer_size": 64,
13
+ "class_num": 1000,
14
+ "decoder_depth": 16,
15
+ "decoder_embed_dim": 1024,
16
+ "decoder_num_heads": 16,
17
+ "diffloss_d": 3,
18
+ "diffloss_w": 1024,
19
+ "diffusion_batch_mul": 4,
20
+ "encoder_depth": 16,
21
+ "encoder_embed_dim": 1024,
22
+ "encoder_num_heads": 16,
23
+ "grad_checkpointing": false,
24
+ "img_size": 256,
25
+ "label_drop_prob": 0.1,
26
+ "mask_ratio_min": 0.7,
27
+ "mlp_ratio": 4.0,
28
+ "model_type": "mar",
29
+ "norm_layer": "LayerNorm",
30
+ "num_sampling_steps": "100",
31
+ "patch_size": 1,
32
+ "proj_dropout": 0.1,
33
+ "transformers_version": "4.44.2",
34
+ "vae_embed_dim": 16,
35
+ "vae_stride": 16
36
+ }