8f382c1ae0417881b806123c9f4546b083ff6a84,experiments/fashion.py,,,#,24

Before 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) = fashion_mnist.load_data()
    x_train = x_train.reshape(x_train.shape + (1,))
    x_test = x_test.reshape(x_test.shape + (1,))

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) = fashion_mnist.load_data()
    x_train = x_train.reshape(x_train.shape + (1,))

Instances