metadata
library_name: segmentation-models-pytorch
license: other
pipeline_tag: image-classification
tags:
- segmentation-models-pytorch
- image-classification
- pytorch
- efficientnet
languages:
- python
Model card for timm-efficientnet-b1.
This repository contains the noisy-student pre-trained weights for the timm-efficientnet-b1 model used as
encoder in the segmentation-models-pytorch library.
Example usage:
- Install the library:
pip install segmentation-models-pytorch
- Use the encoder in your code:
import segmentation_models_pytorch as smp
model = smp.Unet("timm-efficientnet-b1", encoder_weights="noisy-student")