Skip to content

Commit

Permalink
Add norm name arg to Swin UNETR BTCV (#171)
Browse files Browse the repository at this point in the history
  • Loading branch information
lweitkamp authored Jan 4, 2023
1 parent 1dde7a1 commit 07da7ad
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions SwinUNETR/BTCV/main.py
Original file line number Diff line number Diff line change
Expand Up @@ -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")
Expand Down

0 comments on commit 07da7ad

Please sign in to comment.