search

huangyangyi / TeCH

[3DV 2024] Official repo of "TeCH: Text-guided Reconstruction of Lifelike Clothed Humans"
https://huangyangyi.github.io/TeCH/
MIT License
379 stars 24 forks source link

Questions about GPUs #21

Closed glorioushonor closed 8 months ago

glorioushonor commented 8 months ago

In the process of fine-tuning Dreambooth, is it necessary for each GPU to have 32GB of VRAM? I encounter Out-Of-Memory (OOM) errors when running the code on a server with 4 GPUs, each having 24GB of VRAM.

huangyangyi commented 8 months ago

Hi, you can refer to this issue and try adjusting the batchsize

wyiguanw commented 6 months ago

@glorioushonor Hi, have you been able to run on 24G GPUS? if you do, could you let me know how you did that? I tried setting all batch-size to 1 , and adding those arguments in the command line, seems not working.