b115f1f721594772ca12e02dc388b1b210a2ee73,experiments/mnist.py,,,#,31

Before Change

if __name__ == "__main__":
    select_gpu()
    (x_train, y_train), (x_test, y_test) = mnist.load_data()
    X = np.concatenate((x_train, x_test))
    Y = np.concatenate((y_train, y_test))
    clf = ImageClassifier(searcher_type=sys.argv[1], path=sys.argv[2], verbose=False)

After Change

if __name__ == "__main__":
    // select_gpu()
    os.environ["CUDA_VISIBLE_DEVICES"] = str(3)
    constant.LIMIT_MEMORY = True
    (x_train, y_train), (x_test, y_test) = mnist.load_data()
    x_train = x_train.reshape(x_train.shape + (1,))
    x_test = x_test.reshape(x_test.shape + (1,))

Instances