medicaltorch icon indicating copy to clipboard operation
medicaltorch copied to clipboard

Published 20 hours ago • verified publisher icon perone

Custom Datasets can now pin memory

Open perone opened this issue 5 years ago • 0 comments

According to the PR (https://github.com/pytorch/pytorch/pull/16743), it is now possible to pin memory for custom datasets. We should definitely implement that in order to overlap GPU copies and computation.

perone avatar Feb 19 '19 14:02 perone