File size: 103 Bytes
4113c4d
 
1
2
3
import deeprobust.image.netmodels.train_model as trainmodel
trainmodel.train('CNN','MNIST','cuda', 20)