data icon indicating copy to clipboard operation
data copied to clipboard

Published 20 hours ago • verified publisher icon pytorch

Add `pin_memory` Adapter

Open NivekT opened this issue 2 years ago • 0 comments

🚀 The feature

Add an Adapter for pin_memory feature, similar to enable/disable shuffle.

Motivation, pitch

This allows users to enable/disable/change device of PinMemoryIterDataPipe without making changes to the DataPipe graph.

Alternatives

No response

Additional context

No response

NivekT avatar Feb 22 '23 20:02 NivekT