Which GPU do you used in training?

[Yiir-1]

Hello, thanks for the helpful work. I wander to know which GPU do you used in training and how long do you used in training?

[Jeoyal]

Hi @Yiir-1 , thank you for your interest in our work. As mentioned in our paper, we train StyleShot on a single machine with 8 A100 GPUs for 360k steps (300k for stage one, 60k for stage two) with a batch size of 16 per GPU, which takes almost 3 days.