This folder contains 70+ computer vision models. Some are from Imgclsmob. We reimplement some of them, add new APIs (e.g., num_classes), and support them in FedScale benchmarking. Please acknowledge to Imgclsmob if you use any of the model herein.
The full list of supported models are available here. Note that for small images (e.g., FMNIST), we suggest using models with -cifar suffix, as they have smaller kernels and strides. Meanwhile, please ignore the suffix -cifar10 or -cifar100 as their model num_classes will be automatically overrided by the --num_classes setting of the dataset.
We are adding new models, and appreciate if you can consider contributing yours! Please feel free to report bugs.