Skip to content
Snippets Groups Projects
Unverified Commit e206cc59 authored by Alexandru-Mihai GHERGHESCU's avatar Alexandru-Mihai GHERGHESCU
Browse files

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
1 merge request!25Re-factor optimus-prime code (optimus-prime v2)
Loading
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment