ChenRocks / UNITER

Research code for ECCV 2020 paper "UNITER: UNiversal Image-TExt Representation Learning"
https://arxiv.org/abs/1909.11740
777 stars 109 forks source link

When I set num_worker=8, the following error will occurs #58

Open youngfly11 opened 3 years ago

youngfly11 commented 3 years ago

When I try to run the pretrain.py file, and set the work=8, the following error will appear. The training is very slow when the worker=4, the gpu utilization is very low, How do I solve it? thanks image