5ea0c27f524a3e684d9a2e6f012e30e6da2de8aa,scripts/bert/finetune_classifier.py,,test,#Any#Any#,334

Before Change

    logging.info("Now we are doing testing on %s with %s.", segment, ctx)

    tic = time.time()
    value_list = []
    index_list = []
    for _, seqs in enumerate(loader_test):
        input_ids, valid_length, type_ids = seqs
        out = model(input_ids.as_in_context(ctx),

After Change

                results.append("{:.3f}".format(result))
        else:
            // classification task
            indices = mx.nd.topk(out, k=1, ret_typ="indices", dtype="int32").asnumpy()
            for index in indices:
                results.append(task.class_labels[int(index)])

    mx.nd.waitall()
    toc = time.time()
    logging.info("Time cost=%.2fs, throughput=%.2f samples/s", toc - tic,
                 dev_batch_size * len(loader_test) / (toc - tic))

Instances