Can you provide multi-GPU version?

[xmu-xiaoma666]

Can you provide multi-GPU version?

[QinYang79]

Can you provide multi-GPU version?

I'm also experimenting with multiple GPUs. I found that using nn.DataParallel() and modifying some code as follows seem to achieve parallelism.

trainer.py text_embeds, video_features, video_embeds_pooled= self.model(data,True) video_embeds_pooled = self.model.module.pool_frames(text_embeds, video_features) output = sim_matrix_training(text_embeds, video_embeds_pooled, self.pooling_type)

[Jiamian-Wang]

Thanks for the nice and clear implementation! Is that possible to release a multi-GPU version?

[X7J92]

您好呀，我想问一下，您在训练MSVD数据集，特定的迭代次数被系统自动杀死·的情况吗 @Jiamian-Wang

[X7J92]

您好呀，我想问一下，您在训练MSVD数据集，特定的迭代次数被系统自动杀死·的情况吗？ @QinYang79