support models from other packages

[jxchen01]

MMSeg: https://mmsegmentation.readthedocs.io/en/latest/api.html#module-mmseg.models.backbones

https://github.com/rwightman/pytorch-image-models