FFCV: Accelerated Model Training via Fast Data Loading

Đây là một thư viện giúp tăng tốc qúa trình training mà độ chính xác giảm không đáng kể.

Cài đặt:

conda create -n ffcv python=3.9 cupy pkg-config compilers libjpeg-turbo opencv pytorch torchvision cudatoolkit=11.3 numba -c pytorch -c conda-forge && conda activate ffcv && pip install ffcv

Thay thế data loaders hiện có:

Test trên bộ Image Net

@misc{leclerc2022ffcv,
author = {Guillaume Leclerc and Andrew Ilyas and Logan Engstrom and Sung Min Park and Hadi Salman and Aleksander Madry},
title = {ffcv},
year = {2022},
howpublished = {\url{https://github.com/libffcv/ffcv/}},
note = {commit xxxxxxx}
}