huangyangyi / TeCH

[3DV 2024] Official repo of "TeCH: Text-guided Reconstruction of Lifelike Clothed Humans"

Home Page:https://huangyangyi.github.io/TeCH/

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Questions about GPUs

glorioushonor opened this issue · comments

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.

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

@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.