aditya-grover / climate-learn

Source code for ClimateLearn
MIT License
310 stars 49 forks source link

Prefetching for __iter__() in ShardDataset #92

Closed prakhar6sharma closed 1 year ago

prakhar6sharma commented 1 year ago

Is your feature request related to a problem? Please describe. Currently, ShardDataset loads a chunk and only after it exhausts it, it loads the next one. It would be great to hide the latency by implementing some sort of prefetching.

Describe the solution you'd like I have no clue on how to attain it but would love to hear other's thoughts.