KeyError: 'resnet18'

[c1a1o1]

E:\Users\Raytine\Anaconda3\python.exe F:/imageedit/face-attribute-prediction-master/main.py
=> creating model 'resnet18'
Traceback (most recent call last):
File "F:/imageedit/face-attribute-prediction-master/main.py", line 417, in
main()
File "F:/imageedit/face-attribute-prediction-master/main.py", line 135, in main
model = models.dictargs.arch
KeyError: 'resnet18'

[xugangtongji]

there is just res50 had been wrritten in model. Chang main res18 to res50