【pytorch】torch.utils.data.DataLoader中的pin_memory屬性

doc解釋: https://pytorch.org/docs/stable/_modules/torch/utils/data/dataloader.html#DataLoader pin_memory (bool, optional): If True, the data loader will copy tensors into CUDA pinned memory before retur
相關文章
相關標籤/搜索