Function custom_multi_gpu_test

Test model with multiple gpus. This method tests model with multiple gpus and collects the results under two different modes: gpu and cpu modes. By setting 'gpu_collect=True' it encodes results to gpu tensors and use gpu communication for results collection. On cpu mode it saves

(model, data_loader, tmpdir=None, gpu_collect=False)