microsoft / scene_graph_benchmark

image scene graph generation benchmark
MIT License
382 stars 86 forks source link

Why does the program freeze when running with multiple GPUs? #48

Open tao123322 opened 2 years ago

tao123322 commented 2 years ago

python -m torch.distributed.launch --nproc_per_node=$NGPUS tools/train_sg_net.py --config-file "sgg_configs/vg_vrd/rel_danfeiX_FPN50_reldn.yaml"


Setting OMP_NUM_THREADS environment variable for each process to be 1 in default, to avoid your system being overloaded, please further tune the variable for optimal performance in your application as needed.