Add support for Imagenet training
arnaghosh opened this issue · comments
Arna Ghosh commented
- Add FFCV dataloader for Imagenet (with support for multiple augmentations)
- Test imagenet dataloaders for data shape/size
- Test runtime for 1 epoch with ResNet18 model and Adam optimizer
- Run hparam sweep