Hugging Face's logo

gihakkk
/
autoencoder_model.keras

Keras
Model card Files Community
gihakkk commited on
Commit
b175759
•
1 Parent(s): 8fc2bb2

Update config.json

Browse files
Files changed (1) hide show
  1. config.json +87 -31
config.json CHANGED
@@ -1,32 +1,88 @@
1
  {
2
- "_name_or_path": "bert-base-multilingual-cased",
3
- "architectures": [
4
- "BertForSequenceClassification"
5
- ],
6
- "attention_probs_dropout_prob": 0.1,
7
- "classifier_dropout": null,
8
- "directionality": "bidi",
9
- "hidden_act": "gelu",
10
- "hidden_dropout_prob": 0.1,
11
- "hidden_size": 768,
12
- "initializer_range": 0.02,
13
- "intermediate_size": 3072,
14
- "layer_norm_eps": 1e-12,
15
- "max_position_embeddings": 512,
16
- "model_type": "bert",
17
- "num_attention_heads": 12,
18
- "num_hidden_layers": 12,
19
- "pad_token_id": 0,
20
- "pooler_fc_size": 768,
21
- "pooler_num_attention_heads": 12,
22
- "pooler_num_fc_layers": 3,
23
- "pooler_size_per_head": 128,
24
- "pooler_type": "first_token_transform",
25
- "position_embedding_type": "absolute",
26
- "problem_type": "single_label_classification",
27
- "torch_dtype": "float32",
28
- "transformers_version": "4.41.2",
29
- "type_vocab_size": 2,
30
- "use_cache": true,
31
- "vocab_size": 119547
32
- }
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
  {
2
+ "input_shape": [
3
+ 152,
4
+ 152,
5
+ 3
6
+ ],
7
+ "encoder": {
8
+ "conv_layers": [
9
+ {
10
+ "filters": 32,
11
+ "kernel_size": [
12
+ 3,
13
+ 3
14
+ ],
15
+ "activation": "relu"
16
+ },
17
+ {
18
+ "filters": 64,
19
+ "kernel_size": [
20
+ 3,
21
+ 3
22
+ ],
23
+ "activation": "relu"
24
+ },
25
+ {
26
+ "filters": 128,
27
+ "kernel_size": [
28
+ 3,
29
+ 3
30
+ ],
31
+ "activation": "relu"
32
+ }
33
+ ],
34
+ "pooling": {
35
+ "size": [
36
+ 2,
37
+ 2
38
+ ]
39
+ }
40
+ },
41
+ "decoder": {
42
+ "conv_layers": [
43
+ {
44
+ "filters": 128,
45
+ "kernel_size": [
46
+ 3,
47
+ 3
48
+ ],
49
+ "activation": "relu"
50
+ },
51
+ {
52
+ "filters": 64,
53
+ "kernel_size": [
54
+ 3,
55
+ 3
56
+ ],
57
+ "activation": "relu"
58
+ },
59
+ {
60
+ "filters": 32,
61
+ "kernel_size": [
62
+ 3,
63
+ 3
64
+ ],
65
+ "activation": "relu"
66
+ }
67
+ ],
68
+ "upsampling": {
69
+ "size": [
70
+ 2,
71
+ 2
72
+ ]
73
+ },
74
+ "output_layer": {
75
+ "filters": 3,
76
+ "kernel_size": [
77
+ 3,
78
+ 3
79
+ ],
80
+ "activation": "sigmoid"
81
+ }
82
+ },
83
+ "training": {
84
+ "optimizer": "adam",
85
+ "loss": "binary_crossentropy",
86
+ "epochs": 50
87
+ }
88
+ }