qguopku

Results 29 comments of qguopku

Hi, you can use the latest train_hack.py, I will release it once I have trained a robust enough model.

hack是我自己实现的,非hack的是原文的(非常难控制pose,需要大规模数据+overfit)

> > hack是我自己实现的,非hack的是原文的(非常难控制pose,需要大规模数据+overfit) > > 原来如此,感谢大佬,数据量小的话hack的效果更好些是吗? 对,即使数据量大,我也建议使用hack

It is frustrating to note that the current training code requires a minimum of 40G VRAM for the A100.

> @guoqincode hello sir, that means after stage1 training, the unet cannot generate content based on the prompt word? if i set unet.requires_grad_(False) and unet.eval(), will unet still gain the...

I'm sorry, but I found that tricks (FP16, BF16, etc.) that reduce the memory will reduce the performance of the model (NAN loss often occurs), so 80GB VRAM is best...

You can use: https://github.com/IDEA-Research/DWPose

Hello, have you changed the code? I use the same code that is trainable.