not-lain commited on
Commit
6a01b9e
·
verified ·
1 Parent(s): 75fdcfc

Delete base.yaml

Browse files
Files changed (1) hide show
  1. base.yaml +0 -58
base.yaml DELETED
@@ -1,58 +0,0 @@
1
- pixel_mean: [0.485, 0.456, 0.406]
2
- pixel_std: [0.229, 0.224, 0.225]
3
-
4
- pixel_dim: 256
5
- key_dim: 64
6
- value_dim: 256
7
- sensory_dim: 256
8
- embed_dim: 256
9
-
10
- pixel_encoder:
11
- type: resnet50
12
- ms_dims: [1024, 512, 256, 64, 3] # f16, f8, f4, f2, f1
13
-
14
- mask_encoder:
15
- type: resnet18
16
- final_dim: 256
17
-
18
- pixel_pe_scale: 32
19
- pixel_pe_temperature: 128
20
-
21
- object_transformer:
22
- embed_dim: ${model.embed_dim}
23
- ff_dim: 2048
24
- num_heads: 8
25
- num_blocks: 3
26
- num_queries: 16
27
- read_from_pixel:
28
- input_norm: False
29
- input_add_pe: False
30
- add_pe_to_qkv: [True, True, False]
31
- read_from_past:
32
- add_pe_to_qkv: [True, True, False]
33
- read_from_memory:
34
- add_pe_to_qkv: [True, True, False]
35
- read_from_query:
36
- add_pe_to_qkv: [True, True, False]
37
- output_norm: False
38
- query_self_attention:
39
- add_pe_to_qkv: [True, True, False]
40
- pixel_self_attention:
41
- add_pe_to_qkv: [True, True, False]
42
-
43
- object_summarizer:
44
- embed_dim: ${model.object_transformer.embed_dim}
45
- num_summaries: ${model.object_transformer.num_queries}
46
- add_pe: True
47
-
48
- aux_loss:
49
- sensory:
50
- enabled: True
51
- weight: 0.01
52
- query:
53
- enabled: True
54
- weight: 0.01
55
-
56
- mask_decoder:
57
- # first value must equal embed_dim
58
- up_dims: [256, 128, 128, 64, 16]