can i train the retinanet with your code？

tianzhi0549 / FCOS

FCOS: Fully Convolutional One-Stage Object Detection (ICCV'19)

https://arxiv.org/abs/1904.01355

Other

3.27k stars 630 forks source link

can i train the retinanet with your code？ #135

Open qianzhang2018 opened 5 years ago

qianzhang2018 commented 5 years ago

I want to make a comparison on the same code.can i train the retinanet with your code？ can i use this？ python -m torch.distributed.launch --nproc_per_node=1 --master_port=$((RANDOM + 10000)) tools/train_net.py --skip-test --config-file configs/retinanet/retinanet_R-50-FPN_1x.yaml DATALOADER.NUM_WORKERS 2 OUTPUT_DIR training_dir/retinanet？

tianzhi0549 commented 5 years ago

@qianzhang2018 Yes, you can use the command line to train RetinaNet.