search

homangab / Track-2-Act

code for the paper Predicting Point Tracks from Internet Videos enables Diverse Zero-Shot Manipulation
Other
56 stars 4 forks source link

Training cost of track prediction model #1

Closed ziyin-xiong closed 1 month ago

ziyin-xiong commented 3 months ago

Hello! I am currently attempting to reproduce your work on track prediction. I'm wondering if you could kindly share some details of the training costs involved in the track prediction task, such as the number of gpus required, training duration, batch size, and other pertinent information.

hnuzhy commented 1 month ago

I'm also looking forward to know more details of the training cost. For example, how much time will it take if I use 4 A100 (80GB/per) GPUs?

homangab commented 1 month ago

Hello, Thanks for the questions! And apologies for my very late response. For this work, I used 8 A100s (80GB per gpu) with a batch size of 256 and the training duration was around 72 hours.

ziyin-xiong commented 1 month ago

Thanks for your reply!