layer6ai-labs / xpool

https://layer6ai-labs.github.io/xpool/
110 stars 9 forks source link

Doesn't support multi-gpu training, it is not possible to achieve video_features_pooled. #16

Closed RuixiangZhao closed 1 year ago

RuixiangZhao commented 1 year ago

Your code does not support multi-GPU training, especially at the clip_transformer.py#L37 position. Even with the use of gather operations in multi-GPU, it is not possible to achieve video_features_pooled.

NoelVouitsis commented 1 year ago

You are correct that this operation as it stands in the code cannot be done on multi-GPU, but you can certainly still apply CLIP or other operations in the code to a multi-GPU setting. Thanks!