a | b/segmentation/train.sh | ||
---|---|---|---|
1 | # Copyright (c) Microsoft Corporation. All rights reserved. |
||
2 | # Licensed under the MIT License. |
||
3 | |||
4 | torchrun --standalone --nproc_per_node=1 trainddp.py --fold=1 --network-name='unet' --epochs=4 --input-patch-size=192 --train-bs=1 --num_workers=2 --lr=2e-4 --wd=1e-5 --val-interval=2 --sw-bs=2 --cache-rate=1 |