SHI-Labs / Convolutional-MLPs

[Preprint] ConvMLP: Hierarchical Convolutional MLPs for Vision, 2021

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

slight problem

HWalkingMan opened this issue · comments

assert len(blocks) == len(mlp_ratios) == len(mlp_ratios), \

should be

assert len(blocks) == len(dims) == len(mlp_ratios), \
            f"blocks, dims and mlp_ratios must agree in size, {len(blocks)}, {len(dims)} and {len(mlp_ratios)} passed."

Thanks for pointing it out, we've fixed it