IDM-VTON What is the minimum GPU requirement to train this code???

What is the minimum GPU requirement to train this code???

Open kamrul-NSL opened this issue 1 year ago • 7 comments

I want to train this code on my own dataset. What is the minimum GPU requirement??? I am getting Cuda out of memory error on RTX-3090 24GB GPU. TIA

Aug 07 '24 09:08 kamrul-NSL

I also want to know the minimum GPU requirements

Sep 05 '24 04:09 awais-nayyar

about 58G

Sep 14 '24 02:09 TnoobT

@TnoobT What was the batch size? Did you train the model? Full model? On only decoder layer? Thanks for the info.

Sep 18 '24 11:09 kamrul-NSL

@TnoobT What was the batch size? Did you train the model? Full model? On only decoder layer? Thanks for the info.

No changes made, consistent with the training code provided by the author. train tryon net

Sep 18 '24 11:09 TnoobT

@TnoobT can you please share the training results? Did you train on custom dataset? what about the loss? is it decreased?

Sep 19 '24 02:09 arifshahriarnsl

@TnoobT can you please share the training results? Did you train on custom dataset? what about the loss? is it decreased?

The effect is not good, the inpaint is incomplete and there is also color difference. loss: 0.01-0.05，loss not reduced.

Sep 19 '24 02:09 TnoobT

I am getting insufficient memory issue on GPU A100. Am i doing something wrong while training? or the GPU has less memory for its training?

Feb 11 '25 12:02 shahzain110