ShuxianZou commited on
Commit
e916ae9
·
verified ·
1 Parent(s): d3aef1a

Update config.yaml

Browse files
Files changed (1) hide show
  1. config.yaml +15 -13
config.yaml CHANGED
@@ -65,19 +65,21 @@ model:
65
  - dna_seq
66
  - rna_seq
67
  adapter:
68
- fusion:
69
- class_path: modelgenerator.adapters.fusion.ConcatFusion
70
- init_args:
71
- project_size: 1024
72
- pooling: mean_pooling
73
- adapter:
74
- class_path: modelgenerator.adapters.MLPAdapter
75
- init_args:
76
- hidden_sizes:
77
- - 1024
78
- bias: true
79
- dropout: 0.1
80
- dropout_in_middle: false
 
 
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