examples/image/test_train.py

import deeprobust.image.netmodels.train_model as trainmodel
trainmodel.train('CNN','MNIST','cuda', 20)