Skip to content

Commit 29a33ce

Browse files
authored
Update depth parameter name
1 parent 6b79c83 commit 29a33ce

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

Diff for: README.md

+1-1
Original file line numberDiff line numberDiff line change
@@ -150,7 +150,7 @@ mask, label = model(x)
150150
Depth parameter specify a number of downsampling operations in encoder, so you can make
151151
your model lighted if specify smaller `depth`.
152152
```python
153-
model = smp.FPN('resnet34', depth=4)
153+
model = smp.Unet('resnet34', encoder_depth=4)
154154
```
155155

156156

0 commit comments

Comments
 (0)