how to support batch_size > 1

jinxixiang / magic_animate_unofficial

An unoffical training code of Magic Animate

Apache License 2.0

35 stars 5 forks source link

how to support batch_size > 1 #13

Open WilliamZhaoz opened 8 months ago

WilliamZhaoz commented 8 months ago

Thanks for your great efforts!

I'm reading and try to runing the training code, I noticed that, in train.py, there are 2 comments "FIXME: only support train_batch_size=1", I'd like to know would you please to share the training code for batch_size > 1, if not, are there other issues about why the script can only support batch_size=1, if I want to modify it, are there other trap?

Looking forward to your reply. Thanks!

Eurekaarrow commented 2 weeks ago

Same question! Have you solved this "batch_size > 1"? Is there any issues during the training process?