【实验】fcos 遥感

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

你可能感兴趣的:(跑实验,目标检测,深度学习,pytorch,fcos)