From 953226f561b41eab1e22cb7b2824362ec24dcf1a Mon Sep 17 00:00:00 2001 From: "Jefferson F. Silva" Date: Thu, 8 Jun 2023 19:54:24 -0300 Subject: [PATCH] Add new swin transforme networks --- segmentation_models_pytorch/encoders/swin_transformer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/segmentation_models_pytorch/encoders/swin_transformer.py b/segmentation_models_pytorch/encoders/swin_transformer.py index 4bbed48e..daf00146 100644 --- a/segmentation_models_pytorch/encoders/swin_transformer.py +++ b/segmentation_models_pytorch/encoders/swin_transformer.py @@ -692,7 +692,7 @@ def load_state_dict(self, state_dict): def get_pretrained_cfg(name): return { - "url": " https://github.com/jeffersonfs/segmentation_models.pytorch/releases/download/v0.0.1-alpha/{}.pth".format(name), + "url": " https://github.com/jeffersonfs/segmentation_models.pytorch/releases/download/v0.0.2-alpha/{}.pth".format(name), "mean": [0.485, 0.456, 0.406], "std": [0.229, 0.224, 0.225], }