kuhung/SSD_keras

gpu 利用率不高

Closed this issue · 3 comments

你好,我在训练数据时发现GPU的利用率不超过40%,请问能在哪里设置让他的可利用率高一点,以加快训练吗? 十分感谢

@unlikerain 这行代码对 GPU 的显存起限定作用,但他只出现在 SSD_crop.py 中。如果代码没改动的话,不知道你的 GPU 是不是多人在共用。管理员可能对每个人的使用进行了分配限定。

config.gpu_options.per_process_gpu_memory_fraction = 0.8

如果是要加快训练,可增大 batch size

好的,非常感谢