d3e50a9377058fb1e6d7be15f9c00bd6f3173e33,examples/mnist_mlp.py,,main,#Any#Any#Any#,13

Before Change

def main(depth=2, width=512, nb_epoch=30):
    if CupyOps.xp != None:
        Model.ops = CupyOps()
        Model.Ops = CupyOps
    // Configuration here isn"t especially good. But, for demo..
    with Model.define_operators({"**": clone, ">>": chain}):
        model = ReLu(width) >> ReLu(width) >> Softmax()

    train_data, dev_data, _ = datasets.mnist()

After Change

def main(depth=2, width=512, nb_epoch=30):
    prefer_gpu()
    // Configuration here isn"t especially good. But, for demo..
    with Model.define_operators({"**": clone, ">>": chain}):
        model = ReLu(width) >> ReLu(width) >> Softmax()

Instances