Skip to content

Models are missing `model_transforms` property

As of now, models are missing a model_transforms property, one of them being the resizing to the model input size.

I'm not sure how the examples with Pasa, Densenet and Alexnet are working right now, but this is definitely something we want to have around.

The model_transforms provided by the models should be set on our own CachingDataModule.