From 07da7ad36a0b218556a427f3bc5292a772c30f60 Mon Sep 17 00:00:00 2001 From: Laurens Weitkamp Date: Wed, 4 Jan 2023 01:35:07 +0100 Subject: [PATCH] Add norm name arg to Swin UNETR BTCV (#171) --- SwinUNETR/BTCV/main.py | 1 + 1 file changed, 1 insertion(+) diff --git a/SwinUNETR/BTCV/main.py b/SwinUNETR/BTCV/main.py index 7da016d9..666ede3e 100644 --- a/SwinUNETR/BTCV/main.py +++ b/SwinUNETR/BTCV/main.py @@ -59,6 +59,7 @@ parser.add_argument("--rank", default=0, type=int, help="node rank for distributed training") parser.add_argument("--dist-url", default="tcp://127.0.0.1:23456", type=str, help="distributed url") parser.add_argument("--dist-backend", default="nccl", type=str, help="distributed backend") +parser.add_argument("--norm_name", default="instance", type=str, help="normalization name") parser.add_argument("--workers", default=8, type=int, help="number of workers") parser.add_argument("--feature_size", default=48, type=int, help="feature size") parser.add_argument("--in_channels", default=1, type=int, help="number of input channels")