【实验】fcos 遥感

ⅰ亾dé卋堺 提交于 2020-08-11 02:08:31

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

标签
易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!