ShuxianZou commited on
Commit
3a28270
·
verified ·
1 Parent(s): 5e640f5

Update config.yaml

Browse files
Files changed (1) hide show
  1. config.yaml +15 -13
config.yaml CHANGED
@@ -83,19 +83,21 @@ model:
83
  - rna_seq
84
  - protein_seq
85
  adapter:
86
- fusion:
87
- class_path: modelgenerator.adapters.fusion.ConcatFusion
88
- init_args:
89
- project_size: 1024
90
- pooling: mean_pooling
91
- adapter:
92
- class_path: modelgenerator.adapters.MLPAdapter
93
- init_args:
94
- hidden_sizes:
95
- - 1024
96
- bias: true
97
- dropout: 0.1
98
- dropout_in_middle: false
 
 
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