pip install darmo

import darmo

# just change -> "dartsv2", "pdarts", "relative_nas", "nasnet", "eeea_c2", "eeea_c1"
model = darmo.create_model("dartsv2", num_classes=1000, pretrained=True, auxiliary=True)

# create model with ImageNet pretrained
model = darmo.create_model("dartsv2", num_classes=1000, pretrained=True, auxiliary=True)

# Reset classifier layer with add dropout before classifier layer
model.reset_classifier(num_classes=100, dropout=0.2)

git clone https://github.com/jitdee-ai/darmo/
cd darmo
python test_imagenet.py --arch relative_nas --data [path of imagenet include val folder]

@software{chakkrit_termritthikun_2020_4139755,
author       = {Chakkrit Termritthikun},
title        = {jitdee-ai/darmo: pre-trained models for darts},
month        = oct,
year         = 2020,
publisher    = {Zenodo},
version      = {0.0.4},
doi          = {10.5281/zenodo.4139755},
url          = {https://doi.org/10.5281/zenodo.4139755}
}