mobilenetv3细分
的非官方实现,用于语义分割。
必需品
PyTorch 1.1
Python 3.x
用法
火车
单GPU训练
python train.py --model mobilenetv3_small --dataset citys --lr 0.0001 --epochs 240
多GPU训练
# for example, train mobilenetv3 with 4 GPUs:
export NGPUS=4
python -m torch.distributed.launch --nproc_per_node=$NGPUS train.py --model mobilenetv3_small --dataset citys --lr 0.0001 --epochs 240
评估
单GPU训练
python eval.py --model mobilenetv
1