Update config.yaml
Browse files- config.yaml +15 -13
config.yaml
CHANGED
@@ -83,19 +83,21 @@ model:
|
|
83 |
- rna_seq
|
84 |
- protein_seq
|
85 |
adapter:
|
86 |
-
fusion
|
87 |
-
|
88 |
-
|
89 |
-
|
90 |
-
|
91 |
-
|
92 |
-
|
93 |
-
|
94 |
-
|
95 |
-
|
96 |
-
|
97 |
-
|
98 |
-
|
|
|
|
|
99 |
num_outputs: 30
|
100 |
optimizer:
|
101 |
class_path: torch.optim.AdamW
|
|
|
83 |
- rna_seq
|
84 |
- protein_seq
|
85 |
adapter:
|
86 |
+
class_path: modelgenerator.adapters.fusion.MMFusionTokenAdapter
|
87 |
+
init_args:
|
88 |
+
fusion:
|
89 |
+
class_path: modelgenerator.adapters.fusion.ConcatFusion
|
90 |
+
init_args:
|
91 |
+
project_size: 1024
|
92 |
+
pooling: mean_pooling
|
93 |
+
adapter:
|
94 |
+
class_path: modelgenerator.adapters.MLPAdapter
|
95 |
+
init_args:
|
96 |
+
hidden_sizes:
|
97 |
+
- 1024
|
98 |
+
bias: true
|
99 |
+
dropout: 0.1
|
100 |
+
dropout_in_middle: false
|
101 |
num_outputs: 30
|
102 |
optimizer:
|
103 |
class_path: torch.optim.AdamW
|