b115f1f721594772ca12e02dc388b1b210a2ee73,experiments/cifar10.py,,,#,23

Before Change

    (x_train, y_train), (x_test, y_test) = cifar10.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=True)

    clf.fit(x_train, y_train, time_limit=12*60*60)
    // clf.final_fit(x_train, y_train)
    y = clf.evaluate(x_test, y_test)

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) = cifar10.load_data()
    X = np.concatenate((x_train, x_test))
    Y = np.concatenate((y_train, y_test))

Instances