train.py: Actual training code
utils.py: Configs and other training utils
dataset.py: Contains ImageNet class from AdityaKane2001/regnety
Google supported this work by providing Google Cloud credit. Thanks a lot to TRC group for granting TPU access for training these models!
Models are now available at tensorflow/keras/applications/regnet!