Drop Optimus DataLoader, use PyTorch's built-in
Use PyTorch's built-in Dataloader. This should be much easier to work with, and has better support for pre-fetching, memory pinning and other goodies which improve training time.
parent
56be3a00
No related branches found
No related tags found
Loading
Please register or sign in to comment