search

libffcv / ffcv

FFCV: Fast Forward Computer Vision (and other ML workloads!)
https://ffcv.io
Apache License 2.0
2.82k stars 178 forks source link

Add custom traversal order functionality #141

Closed numpee closed 2 years ago

numpee commented 2 years ago

The order argument can receive a value of OrderOption or a custom subclass of TraversalOrder. Value error raised if the given order is not in OrderOption or does not inherit from TraversalOrder.

GuillaumeLeclerc commented 2 years ago

Thank you for the contribution!

I'm about to merge this in the branch we use for v1 instead of master so that we have less testing to do. Is that ok with you ?

numpee commented 2 years ago

Yup, sounds good!