Is this the correct base configuration for ERADIO?

[Riveroad97]

Hello,

I am implementing the model and would like to confirm whether the following configuration corresponds to the "base" version of ERADIO, as used in your paper or experiments:

def eradio_base_224(pretrained=False, **kwargs):
    model = ERADIO(
        depths=[3, 3, 5, 5],
        num_heads=[2, 4, 8, 16],
        window_size=[None, None, [7, 7], 7],
        dim=128,
        in_dim=64,
        mlp_ratio=4,
        drop_path_rate=0.0,
        sr_ratio=[1, 1, [2, 1], 1],
        use_swiglu=False,
        yolo_arch=True,
        shuffle_down=False,
        conv_base=True,
        use_neck=True,
        return_full_features=True,
        full_features_head_dim=1024,
        neck_start_stage=2,
        num_classes=0,
        **kwargs,
    )
    if pretrained:
        model.load_state_dict(torch.load(pretrained)["state_dict"])
    return model

Could you please confirm whether this matches your official eradio-base model setting?
Thanks!