Update examples/semantic-segmentation/run.sh

examples/semantic-segmentation/run.sh

@@ -1,104 +1,76 @@
-# ✅ global correct: 90.9
-# average row correct: ['93.7', '81.9', '80.3', '75.3', '38.0', '57.3', '80.8', '69.2', '88.2', '41.9', '76.2', '63.8', '70.7', '84.8', '88.5', '90.8', '45.1', '68.3', '51.4', '79.6', '74.3']
-# IoU: ['89.7', '67.8', '66.5', '60.2', '34.6', '45.9', '75.4', '55.0', '77.0', '33.8', '64.5', '33.3', '48.0', '71.3', '79.1', '80.7', '27.7', '55.3', '40.1', '72.8', '57.9']
-# mean IoU: 58.9
-torchrun --nproc_per_node=4 train.py\
+
+torchrun --nproc_per_node=1 train.py\
     --workers 4 --dataset coco --data-path /home/cs/Documents/datasets/coco\
     --model fcn_resnet50 --aux-loss --weights FCN_ResNet50_Weights.COCO_WITH_VOC_LABELS_V1\
     --test-only
 
-# ✅ global correct: 91.1
-# average row correct: ['93.9', '82.1', '79.2', '77.1', '37.9', '57.8', '81.9', '69.1', '91.3', '40.5', '80.0', '63.4', '66.9', '84.5', '90.7', '90.8', '44.4', '69.3', '50.5', '84.2', '74.0']
-# IoU: ['89.9', '70.0', '67.0', '61.2', '34.4', '47.0', '74.8', '56.1', '77.5', '33.7', '64.3', '33.5', '49.9', '71.2', '79.6', '81.2', '27.7', '55.3', '40.8', '74.6', '60.5']
-# mean IoU: 59.5
-# Training time 2:40:17
+
 torchrun --nproc_per_node=4 train.py\
     --workers 4 --dataset coco --data-path /home/cs/Documents/datasets/coco\
     --model fcn_resnet50 --aux-loss --output-dir fcn_resnet50 --weights FCN_ResNet50_Weights.COCO_WITH_VOC_LABELS_V1\
     --epochs 5 --batch-size 16 --lr 0.0002 --aux-loss --print-freq 100\
     --lr-warmup-method constant --lr-warmup-epochs 3 --lr-warmup-decay 0. \
     --apply-trp --trp-depths 1 --trp-p 0.1 --trp-lambdas 0.4 0.2 0.1
-torchrun --nproc_per_node=4 train.py\
+torchrun --nproc_per_node=1 train.py\
     --workers 4 --dataset coco --data-path /home/cs/Documents/datasets/coco\
     --model fcn_resnet50 --aux-loss --resume fcn_resnet50/model_4.pth\
     --test-only
 
 
-# ✅ global correct: 91.1
-# average row correct: ['93.6', '83.1', '81.1', '77.1', '43.8', '65.3', '80.0', '69.8', '92.0', '47.9', '82.8', '66.5', '71.5', '85.6', '90.6', '91.2', '44.0', '70.1', '59.3', '86.5', '77.5']
-# IoU: ['89.8', '68.1', '68.9', '61.9', '36.8', '52.5', '76.5', '60.8', '80.8', '38.7', '71.9', '33.2', '52.4', '72.4', '81.4', '82.4', '27.6', '60.1', '42.7', '80.9', '65.7']
-# mean IoU: 62.2
-torchrun --nproc_per_node=4 train.py\
+
+torchrun --nproc_per_node=1 train.py\
     --workers 4 --dataset coco --data-path /home/cs/Documents/datasets/coco\
     --model fcn_resnet101 --aux-loss --weights FCN_ResNet101_Weights.COCO_WITH_VOC_LABELS_V1\
     --test-only
 
-# ✅ global correct: 91.1
-# average row correct: ['93.6', '82.8', '79.6', '78.1', '45.2', '64.2', '82.8', '69.2', '92.9', '46.5', '85.3', '65.8', '69.7', '84.9', '92.5', '91.3', '45.7', '72.0', '55.6', '88.9', '77.9']
-# IoU: ['89.8', '66.9', '69.3', '61.8', '37.3', '52.1', '77.8', '61.0', '80.5', '37.8', '70.4', '32.8', '53.5', '73.4', '81.1', '82.2', '28.8', '60.7', '44.5', '81.9', '66.0']
-# mean IoU: 62.4
-# Training time 3:23:30
+
 torchrun --nproc_per_node=4 train.py\
     --workers 4 --dataset coco --data-path /home/cs/Documents/datasets/coco\
     --model fcn_resnet101 --aux-loss --output-dir fcn_resnet101 --weights FCN_ResNet101_Weights.COCO_WITH_VOC_LABELS_V1\
     --epochs 5 --batch-size 12 --lr 0.0002 --aux-loss --print-freq 100\
     --lr-warmup-method constant --lr-warmup-epochs 3 --lr-warmup-decay 0. \
     --apply-trp --trp-depths 1 --trp-p 0.1 --trp-lambdas 0.4 0.2 0.1
-torchrun --nproc_per_node=4 train.py\
+torchrun --nproc_per_node=1 train.py\
     --workers 4 --dataset coco --data-path /home/cs/Documents/datasets/coco\
     --model fcn_resnet101 --aux-loss --resume fcn_resnet101/model_4.pth\
     --test-only
 
 
-# ✅ global correct: 91.5
-# average row correct: ['93.8', '86.1', '83.2', '81.6', '46.6', '57.3', '88.7', '74.4', '93.9', '50.0', '86.9', '61.4', '81.8', '87.4', '93.5', '92.1', '48.7', '89.4', '63.6', '89.1', '73.5']
-# IoU: ['90.2', '68.6', '74.6', '60.2', '41.0', '43.6', '85.3', '54.3', '86.1', '39.3', '81.2', '33.0', '59.0', '71.4', '82.6', '82.7', '25.7', '74.0', '48.6', '75.6', '62.9']
-# mean IoU: 63.8
-torchrun --nproc_per_node=4 train.py\
+
+torchrun --nproc_per_node=1 train.py\
     --workers 4 --dataset coco --data-path /home/cs/Documents/datasets/coco\
     --model deeplabv3_resnet50 --aux-loss --weights DeepLabV3_ResNet50_Weights.COCO_WITH_VOC_LABELS_V1\
     --test-only
 
 
-# ✅ global correct: 91.6
-# average row correct: ['94.1', '85.8', '82.4', '81.5', '44.5', '56.6', '87.3', '73.0', '94.2', '48.2', '87.6', '60.4', '80.1', '87.9', '93.3', '92.0', '47.6', '88.2', '62.9', '88.5', '73.9']
-# IoU: ['90.4', '70.0', '74.6', '62.1', '40.5', '45.2', '83.9', '55.5', '85.7', '39.1', '81.4', '32.6', '60.9', '71.9', '82.6', '82.9', '27.4', '73.5', '47.4', '79.9', '63.7']
-# mean IoU: 64.3
-# # Training time 3:38:30
+
 torchrun --nproc_per_node=4 train.py\
     --workers 4 --dataset coco --data-path /home/cs/Documents/datasets/coco\
     --model deeplabv3_resnet50 --aux-loss --output-dir deeplabv3_resnet50 --weights DeepLabV3_ResNet50_Weights.COCO_WITH_VOC_LABELS_V1\
     --epochs 5 --batch-size 16 --lr 0.0002 --aux-loss --print-freq 100\
     --lr-warmup-method constant --lr-warmup-epochs 3 --lr-warmup-decay 0. \
     --apply-trp --trp-depths 1 --trp-p 0.1 --trp-lambdas 0.4 0.2 0.1
-torchrun --nproc_per_node=4 train.py\
+torchrun --nproc_per_node=1 train.py\
     --workers 4 --dataset coco --data-path /home/cs/Documents/datasets/coco\
     --model deeplabv3_resnet50 --aux-loss --resume deeplabv3_resnet50/model_4.pth\
     --test-only
 
 
-# ✅ global correct: 91.7
-# average row correct: ['94.0', '83.0', '82.9', '82.2', '44.7', '61.0', '86.9', '70.3', '93.6', '54.4', '84.7', '60.5', '82.2', '90.0', '92.4', '92.3', '49.1', '91.8', '66.0', '90.0', '79.7']
-# IoU: ['90.4', '71.3', '74.8', '58.2', '38.9', '49.7', '82.9', '61.9', '86.1', '42.9', '79.1', '32.8', '67.7', '75.1', '81.6', '83.3', '26.9', '74.8', '49.7', '76.4', '66.9']
-# mean IoU: 65.3
-torchrun --nproc_per_node=4 train.py\
+
+torchrun --nproc_per_node=1 train.py\
     --workers 4 --dataset coco --data-path /home/cs/Documents/datasets/coco\
     --model deeplabv3_resnet101 --aux-loss --weights DeepLabV3_ResNet101_Weights.COCO_WITH_VOC_LABELS_V1\
     --test-only
 
 
-# ✅ global correct: 91.8
-# average row correct: ['94.2', '81.6', '82.2', '81.1', '43.2', '60.8', '86.4', '70.6', '94.4', '52.2', '85.1', '60.0', '82.7', '90.3', '92.9', '92.3', '47.6', '90.5', '66.1', '89.7', '77.9']
-# IoU: ['90.5', '72.6', '74.7', '60.9', '37.9', '49.6', '82.6', '62.4', '86.9', '41.9', '79.5', '32.6', '68.4', '75.3', '82.1', '83.3', '28.4', '74.3', '50.5', '77.4', '67.7']
-# mean IoU: 65.7
-# Training time 4:21:32
+
 torchrun --nproc_per_node=4 train.py\
     --workers 4 --dataset coco --data-path /home/cs/Documents/datasets/coco\
     --model deeplabv3_resnet101 --aux-loss --output-dir deeplabv3_resnet101 --weights DeepLabV3_ResNet101_Weights.COCO_WITH_VOC_LABELS_V1\
     --epochs 5 --batch-size 12 --lr 0.0002 --aux-loss --print-freq 100\
     --lr-warmup-method constant --lr-warmup-epochs 3 --lr-warmup-decay 0. \
     --apply-trp --trp-depths 1 --trp-p 0.1 --trp-lambdas 0.4 0.2 0.1
-torchrun --nproc_per_node=4 train.py\
+torchrun --nproc_per_node=1 train.py\
     --workers 4 --dataset coco --data-path /home/cs/Documents/datasets/coco\
     --model deeplabv3_resnet101 --aux-loss --resume deeplabv3_resnet101/model_4.pth\
     --test-only