readme typo (#16)

isaaccorley · Mar 20, 2024 · 86cec24 · 86cec24
1 parent 226250d
commit 86cec24
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/README.md b/README.md
@@ -37,7 +37,7 @@ import torchseg
 model = torchseg.Unet(
     encoder_name="resnet50",
     encoder_weights=True,
-    in_channels=3
+    in_channels=3,
     classes=3,
 )
 ```
@@ -50,7 +50,7 @@ You can also define a `functools.partial` callable as an activation/normalizatio
 model = torchseg.Unet(
     encoder_name="resnet50",
     encoder_weights=True,
-    in_channels=3
+    in_channels=3,
     classes=3,
     encoder_params={
       "act_layer": "prelu",