modules/pytorch_jacinto_ai/xreferences/segmentation/README.md

   1 # Semantic segmentation reference training scripts
   2
   3 This folder contains reference training scripts for semantic segmentation.
   4 They serve as a log of how to train specific models, as provide baseline
   5 training and evaluation scripts to quickly bootstrap research.
   6
   7 All models have been trained on 8x V100 GPUs.
   8
   9 ## fcn_resnet50
  10 ```
  11 python -m torch.distributed.launch --nproc_per_node=8 --use_env train.py --lr 0.02 --dataset coco -b 4 --model fcn_resnet50 --aux-loss
  12 ```
  13
  14 ## fcn_resnet101
  15 ```
  16 python -m torch.distributed.launch --nproc_per_node=8 --use_env train.py --lr 0.02 --dataset coco -b 4 --model fcn_resnet101 --aux-loss
  17 ```
  18
  19 ## deeplabv3_resnet50
  20 ```
  21 python -m torch.distributed.launch --nproc_per_node=8 --use_env train.py --lr 0.02 --dataset coco -b 4 --model deeplabv3_resnet50 --aux-loss
  22 ```
  23
  24 ## deeplabv3_resnet101
  25 ```
  26 python -m torch.distributed.launch --nproc_per_node=8 --use_env train.py --lr 0.02 --dataset coco -b 4 --model deeplabv3_resnet101 --aux-loss
  27 ```