xvjiarui / GCNet

GCNet: Non-local Networks Meet Squeeze-Excitation Networks and Beyond
Apache License 2.0
1.19k stars 165 forks source link

how to reproduce the results of coco detection with only 4 gpus #13

Closed zhaoguangxiang closed 4 years ago

zhaoguangxiang commented 5 years ago

I have 4 tesla v100 gpus, can you recommend the script to reproduce the results

xvjiarui commented 5 years ago

Sorry for the late reply.

You could change imgs_per_gpu=4 in your config file.