def create_model(args):
    model = models.create(args.arch, num_features=args.features, norm=False, dropout=args.dropout, num_classes=args.nclass)
    model.cuda()
    model = nn.DataParallel(model)
    return model
def create_model(args):
    model = models.create(args.arch, num_features=args.features, norm=False, dropout=args.dropout, num_classes=args.nclass)
    model.cuda()
    model = nn.DataParallel(model)
    return model