b61667d3daac5610cf40a7346efddb34476c059b,scripts/bert/finetune_squad.py,,train,#,285

Before Change

                step_loss = 0.0
                log_num = 0
                if args.debug:
                    log.info("Exit early in test mode")
                    break
        epoch_toc = time.time()
        log.info("Time cost={:.2f} s, Thoughput={:.2f} samples/s".format(
            epoch_toc - epoch_tic, total_num/(epoch_toc - epoch_tic)))

After Change

    else:
        train_data = SQuAD(segment, version="1.1")
    if args.debug:
        sampled_data = [train_data[i] for i in range(1000)]
        train_data = mx.gluon.data.SimpleDataset(sampled_data)
    log.info("Number of records in Train data:{}".format(len(train_data)))

    train_data_transform, _ = preprocess_dataset(
        train_data, SQuADTransform(

Instances