How to gather predict on ddp
Problem
I encountered some questions about ddp. Because I train mode with ddp on 2 gpus.
And when i test and predict test dataloader on test_step(), the predict result just half data be predicted.
ex:
original test data: 10000
predict result : 5000 (1/gpus)

How to solve it? use all_gather()?
Hope someone can answer, I will thanks a lot.