fe7d5e9b8bd221685749146e9b42cd44f912891c,examples/mxnet/sampling/dis_sampling/gcn_trainer.py,,main,#Any#,113

Before Change

    // initialize graph
    dur = []
    total_count = 153
    for epoch in range(args.n_epochs):
        for subg_count in range(total_count):
            print(subg_count)
            nf = receiver.recv(g)
            nf.copy_from_parent()
            // forward
            with mx.autograd.record():
                pred = model(nf)

After Change

    for epoch in range(args.n_epochs):
        idx = 0
        for nf in sampler:
            print("epoch: %d, subgraph: %d" %(epoch, idx))
            idx += 1
            nf.copy_from_parent()
            // forward

Instances