Warn users when they are using Pytorch augmentations on the CPU

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

Warn users when they are using Pytorch augmentations on the CPU #127

Closed GuillaumeLeclerc closed 2 years ago

GuillaumeLeclerc commented 2 years ago

It seems that users have tendency to use slow Pytorch augmentations on the CPU with FFCV and expected good performance. We should issue a warning and suggest they replace them with an FFCV augmentation or move it on the GPU