for stage1_layer in self.stage1_layers: # 1
stage1_output, omic_wsi_mgca = stage1_layer(wsi_feats_embed, omic_feats_embed, attn_mask=None)
Stage 2: stage1 fusion embedding and WSI feature fusion again
for stage2_layer in self.stage2_layers: # 2
stage2_output, wsi_omic_mgca = stage2_layer(stage1_output.unsqueeze(0).unsqueeze(0), wsi_feats_embed, attn_mask=None)
`
According to your framework, the second for loop has two layers, S2=2. But from the code perspective, it seems that we can only use the result of the second iteration, so what is the significance of doing the first loop of S2?
`
Stage 1: WSI features and Genomic features fusion
Stage 2: stage1 fusion embedding and WSI feature fusion again
` According to your framework, the second for loop has two layers, S2=2. But from the code perspective, it seems that we can only use the result of the second iteration, so what is the significance of doing the first loop of S2?