Basically in the config files.
Refer to the Reference
CUDA_VISIBLE_DEVICES=4,5,6,7 python train_net_stand.py --config-file configs/Fvlm_coco.yaml --num-gpus 4 (--eval-only)
Pytorch version of F-VLM (ICLR2023)
Model | link | APr | AP |
---|---|---|---|
F-VLM | model | 28.0 | 39.6 |
Reproduce(RN50-coco) | model | 26.3 | 41.0 |