Recursively detach lists of tensors in b ; put them on the CPU if cpu=True . gather only applies during distributed training and the result tensor will be the ...
確定! 回上一頁