Custom Datasets can now pin memory #17
Description
According to the PR (pytorch/pytorch#16743), it is now possible to pin memory for custom datasets. We should definitely implement that in order to overlap GPU copies and computation.
{{ message }}
According to the PR (pytorch/pytorch#16743), it is now possible to pin memory for custom datasets. We should definitely implement that in order to overlap GPU copies and computation.