3a770cbc97085c2cd4eaa0a46b2bc037f35389c2,utils.py,,evaluate_accuracy,#Any#Any#Any#,77

Before Change

        data, label = _get_batch(batch, ctx)
        output = net(data)
        acc += accuracy(output, label)
    return acc / (i+1)

def train(train_data, test_data, net, loss, trainer, ctx, num_epochs, print_batches=None):
    Train a network
    for epoch in range(num_epochs):

After Change

    for batch in data_iterator:
        data, label, batch_size = _get_batch(batch, ctx)
        for X, y in zip(data, label):
            acc += nd.sum(net(X).argmax(axis=1)==y).copyto(mx.cpu())
        acc.wait_to_read() // don"t push too many operators into backend
        n += batch_size
    return acc.asscalar() / n

Instances