Add support for pytorch-segmentation-models

We should consider adding support for the pytorch-segmentation-models library in this package. It has a bunch of implemented models we can re-use in our benchmarks.