【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
相关文章
相关标签/搜索