How to reduce gpus to train?

[Richard-wang85]

Hello! Thank you for your splendid work! But I have some problem on training. I know you train your model with 8 gpus, however, I only have 3 gpus. Could you tell me how to reduce gpus for training! I will appreciate it if you could give me your advise!

[pfeducode]

你好！感谢您的出色工作！但是我在训练上有一些问题。我知道你用 8 个 gpus 训练你的模型，但是，我只有 3 个 gpus。你能告诉我如何减少GPU进行训练！如果您能给我您的建议，我将不胜感激！

you can use argsment parameters --gpu_id and --device_counts to assign your program .it works well

[Richard-wang85]

你好！感谢您的出色工作！但是我在训练上有一些问题。我知道你用 8 个 gpus 训练你的模型，但是，我只有 3 个 gpus。你能告诉我如何减少GPU进行训练！如果您能给我您的建议，我将不胜感激！

you can use argsment parameters --gpu_id and --device_counts to assign your program .it works well

Thank you for your reply! But could you successfully run test_frontal.py?

[pfeducode]

你好！感谢您的一些训练！给我您的建议，我将不胜感激！

您可以使用 argsment 参数 --gpu_id 和 --device_counts 来分配您的程序。它运行良好

感谢你的回复！但是你能成功运行 test_frontal.py 吗？

no, I can't, There are some problems