Skip to content

Commit 86cec24

Browse files
authored
readme typo (#16)
1 parent 226250d commit 86cec24

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

README.md

+2-2
Original file line numberDiff line numberDiff line change
@@ -37,7 +37,7 @@ import torchseg
3737
model = torchseg.Unet(
3838
encoder_name="resnet50",
3939
encoder_weights=True,
40-
in_channels=3
40+
in_channels=3,
4141
classes=3,
4242
)
4343
```
@@ -50,7 +50,7 @@ You can also define a `functools.partial` callable as an activation/normalizatio
5050
model = torchseg.Unet(
5151
encoder_name="resnet50",
5252
encoder_weights=True,
53-
in_channels=3
53+
in_channels=3,
5454
classes=3,
5555
encoder_params={
5656
"act_layer": "prelu",

0 commit comments

Comments
 (0)