search

luxiaolili / IDM-VTON-train

33 stars 3 forks source link

question about memory #2

Open wangzhen-ing opened 1 month ago

wangzhen-ing commented 1 month ago

Hello， thanks for sharing the training code of IDM-VTON. I use a single card 80g A100 for training, batch_size is 1, but it is still out of memory, do you know why? What is the minimum memory required to train this task?

luxiaolili commented 1 month ago

Hello， thanks for sharing the training code of IDM-VTON. I use a single card 80g A100 for training, batch_size is 1, but it is still out of memory, do you know why? What is the minimum memory required to train this task?

I train it use A800 80G, not use 8bit adam train it need 70G，you can add --use_8bit_adam to train.sh