Question about training time and GPU specs

[dhsoon02]

Hello,

Thank you for sharing your excellent research. I really enjoyed reading your work.

I have a quick question regarding the training setup. Could you please let me know the training time for Stage 1 and Stage 2, respectively? I would also appreciate it if you could share the GPU specifications used for training.

Thank you in advance for your help.

[Tsukasane]

Hi, thanks for your interest.

Stage 1 is trained on a single NVIDIA 24GB RTX4090 for approximately 2 days (hasn't supported parallel training for this stage yet). Stage 2 supports parallel training, which we trained on 2x 4090 for approximately 3 days. Training epochs are as reported in the paper.

Please note that the above setting is for max_person=3. A larger maximum group size will lead to more GPU occupancy.

[Threedv]

Hi, thank you for the detailed explanation!

I was wondering whether you would be willing to share the pretrained checkpoint for research purposes. It would be very helpful for reproducing the results and further understanding your method.

Thank you again for your time and support.

Best regards,
Bob