Multi-GPU update with custom collate function to allow variable size target vector per image without needing to pad targets.