Added upsample_kwargs argument to UNetDecoder

BloodAxe · Jan 9, 2024 · 7014381 · 7014381
1 parent 504259e
commit 7014381
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pytorch_toolbelt/modules/decoders/unet.py b/pytorch_toolbelt/modules/decoders/unet.py
@@ -1,4 +1,4 @@
-from typing import List, Union, Type, Tuple
+from typing import List, Union, Type, Tuple, Mapping
 
 import torch
 import logging