search

Hangz-nju-cuhk / Rotate-and-Render

Code for Rotate-and-Render: Unsupervised Photorealistic Face Rotation from Single-View Images (CVPR 2020)
Creative Commons Attribution 4.0 International
489 stars 112 forks source link

How to reduce gpus to train? #43

Open Richard-wang85 opened 2 years ago

Richard-wang85 commented 2 years ago

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 commented 2 years ago

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

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

Richard-wang85 commented 2 years ago

你好！感谢您的出色工作！但是我在训练上有一些问题。我知道你用 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 commented 2 years ago

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

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

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

no, I can't, There are some problems