ICTMCG / Make-Your-Anchor

[CVPR 2024] Make-Your-Anchor: A Diffusion-based 2D Avatar Generation Framework.
327 stars 20 forks source link

About training requirement and duration #5

Closed CHNxindong closed 7 months ago

CHNxindong commented 7 months ago

Hi authors, thanks for your great work. I have a question about training requirement and duration:

The 'Implementation Details' section mentions that all experiments were conducted using a single A100 40G GPU. Would you confirm wheather the training process, when applied to a dataset comprising 27 hours of data collected from four identities, would require only one A100 40G GPU, albeit with an extended duration of seven days?

bone-11 commented 7 months ago

Thank you for your attention. Our approach adopts frame-level training and a training-free strategy for video-level inference. Therefore, the batch size for training the model is set to 4, with 4 images used per training step, and frame-to-frame relationships are not considered. This makes training feasible on a single A100 40G GPU.

CHNxindong commented 7 months ago

I see it. Thanks for your kind response!