Dump model predictions to a pickle file for offline evaluation.
Parameters
out_file_path (str) – Path of the dumped file. Must end with ‘.pkl’
or ‘.pickle’.
collect_device (str) – Device name used for collecting results from
different ranks during distributed training. Must be ‘cpu’ or
‘gpu’. Defaults to ‘cpu’.