CogVideo icon indicating copy to clipboard operation
CogVideo copied to clipboard

get blurry image after lora finetune

Open hanwenxu1 opened this issue 1 year ago • 0 comments

I'm trying to use train_cogvideox_lora.py to finetune 2b model with customized dataset. When I set the train lr is 1e-5, and the epoch is 2, I get some normal result. image Then I found some advice that the lr of lora should be larger, like 5e-4 to 1e-3. I tried to set lr as 5e-4 and the epoch is 10. But I get worse results. image image I wonder that why it became worse and it seemed so vague. The train loss is here: image

hanwenxu1 avatar Oct 31 '24 07:10 hanwenxu1