haochenheheda/Training-Code-of-STM

how to implement multi-gpu for training?

Opened this issue · 0 comments

thanks for your sharing.
but how can i implement multi-gpu for training?and how to raise batchsize to more than 1?
thanks !