search

libffcv / ffcv

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

Fix allocation query for decoding images #366

Open erow opened 4 months ago

erow commented 4 months ago

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

365