c9e80b3f0d81a69e7022b2447ee6809d4e2d2ed7,scripts/bert/finetune_classifier.py,,,#,495

Before Change

if __name__ == "__main__":
    pool_type = os.environ.get("MXNET_GPU_MEM_POOL_TYPE", "")
    if pool_type.lower() == "round":
        logging.info(
            "Setting MXNET_GPU_MEM_POOL_TYPE="Round" may lead to higher memory "
            "usage and faster speed. If you encounter OOM errors, please unset "
            "this environment variable.")
    if not args.only_inference:
        train(task.get_metric())
    elif model_parameters:
        inference(task.get_metric())

After Change

if __name__ == "__main__":
    train(task.metrics)

Instances