qubvel-hf HF staff commited on
Commit
8feb162
·
verified ·
1 Parent(s): da89b84

Upload folder using huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -30,7 +30,7 @@ pip install segmentation-models-pytorch
30
  ```python
31
  import segmentation_models_pytorch as smp
32
 
33
- model = smp.Unet(mobileone_s3, encoder_weights="imagenet")
34
  ```
35
 
36
  ### References
 
30
  ```python
31
  import segmentation_models_pytorch as smp
32
 
33
+ model = smp.Unet("mobileone_s3", encoder_weights="imagenet")
34
  ```
35
 
36
  ### References