rpg_event_representation_learning icon indicating copy to clipboard operation
rpg_event_representation_learning copied to clipboard

Published 20 hours ago • verified publisher icon uzh-rpg

Question about multi-GPU training

Open lisiqi19971013 opened this issue 4 years ago • 0 comments

Hi,

Great work! Thanks for sharing your code. Due to the data format is not a standard mini-batch tensor format as BatchID x Channel x Width x Height, I am wondering whether the model can be trained on multi-GPU? If can, how should I implement it?

Looking forward to your reply, any help would be appreciated. Thanks a lot!

Best wishes,

Siqi Li

lisiqi19971013 avatar Jan 09 '21 04:01 lisiqi19971013