torch.load() uses Python's unpickling facilities but treats storages, which underlie tensors, ... map_location={'cuda:1':'cuda:0'}) # Load tensor from io.
確定! 回上一頁