search

Vchitect / Vlogger

[CVPR2024] Make Your Dream A Vlog
Apache License 2.0
411 stars 42 forks source link

About training #16

Open EugeneCh1a opened 5 months ago

EugeneCh1a commented 5 months ago

I would like to know how many G is required for training. Other than that, I really like showmaker, and his Twist

zhuangshaobin commented 5 months ago

We use an A100 with 80G of video memory for training, and the batchsize of each unit is 3, but I estimate that a single unit's memory can reach 48G to support training that does not require model parallelism.