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
|