Mix ViT in_channels restriction

What's the reasoning behind limiting the Mix Visual Transformer encoder #632 to 3 input channels?

https://github.com/qubvel/segmentation_models.pytorch/blob/master/segmentation_models_pytorch/encoders/mix_transformer.py#L468

I couldn't spot anything in the paper or the original SegFormer implementation.

[github-actions]

This issue is stale because it has been open 60 days with no activity. Remove stale label or comment or this will be closed in 7 days.

[lucasc896]

I've removed the restriction and the model appears to learn. Is there any argument for this restriction?

[github-actions]

This issue is stale because it has been open 60 days with no activity. Remove stale label or comment or this will be closed in 7 days.

[github-actions]

This issue was closed because it has been stalled for 7 days with no activity.

[qubvel]

Would appreciate a PR to fix it! It's basically to enable changing the number of channels in the first conv layer