From 86cec244e261ebcbb7031f826c5bc3e9c8a3d12b Mon Sep 17 00:00:00 2001
From: Nils Lehmann <35272119+nilsleh@users.noreply.github.com>
Date: Wed, 20 Mar 2024 14:24:00 +0100
Subject: [PATCH] readme typo (#16)

---
 README.md | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/README.md b/README.md
index ace89c81..7f9027e0 100644
--- 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",