krasserm commited on
Commit
9d45ef8
·
1 Parent(s): 4b407c6

Upload PerceiverImageClassifier

Browse files
Files changed (2) hide show
  1. config.json +75 -0
  2. pytorch_model.bin +3 -0
config.json ADDED
@@ -0,0 +1,75 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "architectures": [
3
+ "PerceiverImageClassifier"
4
+ ],
5
+ "id2label": {
6
+ "0": 0,
7
+ "1": 1,
8
+ "2": 2,
9
+ "3": 3,
10
+ "4": 4,
11
+ "5": 5,
12
+ "6": 6,
13
+ "7": 7,
14
+ "8": 8,
15
+ "9": 9
16
+ },
17
+ "label2id": {
18
+ "0": 0,
19
+ "1": 1,
20
+ "2": 2,
21
+ "3": 3,
22
+ "4": 4,
23
+ "5": 5,
24
+ "6": 6,
25
+ "7": 7,
26
+ "8": 8,
27
+ "9": 9
28
+ },
29
+ "model_config": {
30
+ "activation_checkpointing": false,
31
+ "activation_offloading": false,
32
+ "decoder": {
33
+ "cross_attention_residual": true,
34
+ "cross_attention_widening_factor": 1,
35
+ "dropout": 0.1,
36
+ "freeze": false,
37
+ "init_scale": 0.1,
38
+ "num_classes": 10,
39
+ "num_cross_attention_heads": 1,
40
+ "num_cross_attention_qk_channels": null,
41
+ "num_cross_attention_v_channels": null,
42
+ "num_output_queries": 1,
43
+ "num_output_query_channels": 128
44
+ },
45
+ "encoder": {
46
+ "cross_attention_widening_factor": 1,
47
+ "dropout": 0.1,
48
+ "first_cross_attention_layer_shared": false,
49
+ "first_self_attention_block_shared": false,
50
+ "freeze": false,
51
+ "image_shape": [
52
+ 28,
53
+ 28,
54
+ 1
55
+ ],
56
+ "init_scale": 0.1,
57
+ "num_cross_attention_heads": 1,
58
+ "num_cross_attention_layers": 2,
59
+ "num_cross_attention_qk_channels": null,
60
+ "num_cross_attention_v_channels": null,
61
+ "num_frequency_bands": 32,
62
+ "num_self_attention_blocks": 3,
63
+ "num_self_attention_heads": 8,
64
+ "num_self_attention_layers_per_block": 3,
65
+ "num_self_attention_qk_channels": null,
66
+ "num_self_attention_v_channels": null,
67
+ "self_attention_widening_factor": 1
68
+ },
69
+ "num_latent_channels": 128,
70
+ "num_latents": 32
71
+ },
72
+ "model_type": "perceiver-io-image-classifier",
73
+ "torch_dtype": "float32",
74
+ "transformers_version": "4.28.0"
75
+ }
pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4338dd5e66d183e4ee6b27010726cb1c5e4ed3e6d87b0d98fb5525dbe02292d0
3
+ size 4087353