First of all, thank you for your great research.
I checked the code, and I would like to confirm: if I set the configuration to ImageNet, does that mean the model can be trained on ImageNet as is?
Also, I noticed that currently only a YAML file for CIFAR-10 is provided in the config.
Could you please share a YAML file or configuration for ImageNet as well, if available?