@@ -17,7 +17,7 @@ class ImageEncoder(nn.Module):
self.encoder = build_neck(cfg, feats_dim[-1], feats_dim[-1])
## CSFM
- self.csfm = build_fpn(cfg=cfg, in_dims=feats_dim, out_dim=round(cfg['d_model']*cfg['width']))
+ self.csfm = build_fpn(cfg=cfg, in_dims=feats_dim, out_dim=round(cfg['d_model']*cfg['width']), input_proj=True)
def forward(self, x):