fcos
数据集
数据集:DOTA1.0_task2_split(fcos跑的是转成coco后的格式)
数据集所在位置:/home/flora/FCOS/datasets/dota
修改了/home/flora/FCOS/fcos_core/config/paths_catalog.py的coco数据集的路径
运行命令
用4张卡训练的
python -m torch.distributed.launch \
--nproc_per_node=4 \
--master_port=$((RANDOM + 10000)) \
tools/train_net.py \
--config-file configs/fcos/fcos_imprv_R_50_FPN_1x.yaml \
DATALOADER.NUM_WORKERS 0 \
OUTPUT_DIR training_dir/fcos_imprv_R_50_FPN_1x
DATALOADER.NUM_WORKERS设置成2不行,必须设成0.
训练时间
开始:2020/8/4 20:32
来源:oschina
链接:https://my.oschina.net/u/4267090/blog/4470111