qubvel-hf HF Staff commited on
Commit
1d86862
·
verified ·
1 Parent(s): bea8b0f

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(timm-efficientnet-b0, encoder_weights="noisy-student")
34
  ```
35
 
36
  ### References
 
30
  ```python
31
  import segmentation_models_pytorch as smp
32
 
33
+ model = smp.Unet("timm-efficientnet-b0", encoder_weights="noisy-student")
34
  ```
35
 
36
  ### References