About GPU and training time

fangyuanmao / PID

Code for PID: Physics-Informed Diffusion Model for Infrared Image Generation

MIT License

43 stars 3 forks source link

About GPU and training time #2

Open chaunceywx opened 1 month ago

chaunceywx commented 1 month ago

I want to ask what platform do you used to train your models? 8*A100? And how long to train this model completely with KAIST or FLIR?

fangyuanmao commented 1 month ago

Thanks for your question. We use 4*V100, with a batchsize=48, accumulate grad=2. The whole training time of model completely on KAIST is 110 hours without testing and 120 hours on FLIR.