diff --git a/README.md b/README.md index 9c9c3ef..4e19a26 100644 --- a/README.md +++ b/README.md @@ -2,7 +2,7 @@ -![Proposed Model](./images/Dyna.png) +![Proposed Model](./images/CAIF.png) ## Updates diff --git a/images/CAIF.png b/images/CAIF.png new file mode 100644 index 0000000..f88aae2 Binary files /dev/null and b/images/CAIF.png differ diff --git a/networks/DynaFormer.py b/networks/DynaFormer.py index 6f5c1f8..2416c91 100644 --- a/networks/DynaFormer.py +++ b/networks/DynaFormer.py @@ -1262,7 +1262,7 @@ def forward(self, x1, x2=None): else: return self.upsample(fused) -class DAEFormer(nn.Module): +class DynaFormer(nn.Module): def __init__(self, num_classes=9, head_count=1, token_mlp_mode="mix_skip"): super().__init__()