Update config.yaml
Browse files- config.yaml +15 -13
config.yaml
CHANGED
@@ -65,19 +65,21 @@ model:
|
|
65 |
- dna_seq
|
66 |
- rna_seq
|
67 |
adapter:
|
68 |
-
fusion
|
69 |
-
|
70 |
-
|
71 |
-
|
72 |
-
|
73 |
-
|
74 |
-
|
75 |
-
|
76 |
-
|
77 |
-
|
78 |
-
|
79 |
-
|
80 |
-
|
|
|
|
|
81 |
num_outputs: 30
|
82 |
optimizer:
|
83 |
class_path: torch.optim.AdamW
|
|
|
65 |
- dna_seq
|
66 |
- rna_seq
|
67 |
adapter:
|
68 |
+
class_path: modelgenerator.adapters.fusion.MMFusionTokenAdapter
|
69 |
+
init_args:
|
70 |
+
fusion:
|
71 |
+
class_path: modelgenerator.adapters.fusion.ConcatFusion
|
72 |
+
init_args:
|
73 |
+
project_size: 1024
|
74 |
+
pooling: mean_pooling
|
75 |
+
adapter:
|
76 |
+
class_path: modelgenerator.adapters.MLPAdapter
|
77 |
+
init_args:
|
78 |
+
hidden_sizes:
|
79 |
+
- 1024
|
80 |
+
bias: true
|
81 |
+
dropout: 0.1
|
82 |
+
dropout_in_middle: false
|
83 |
num_outputs: 30
|
84 |
optimizer:
|
85 |
class_path: torch.optim.AdamW
|