libffcv / ffcv

FFCV: Fast Forward Computer Vision (and other ML workloads!)

https://ffcv.io

Apache License 2.0

2.87k stars 179 forks source link

Closed erow closed 1 month ago

erow commented 9 months ago

This pull reduces the memory usage. Now it allocates a max shape to decode the image.

365