2dd2efb5475cb42a21f0dacaef786c89869bf2de,ml_perf/train_loop.py,,train,#Any#,193

Before Change

    model_path = os.path.join(FLAGS.model_dir, state.train_model_name)

    await checked_run(
        "python3", "train.py",
        "--flagfile={}".format(os.path.join(FLAGS.flags_dir, "train.flags")),
        "--work_dir={}".format(FLAGS.work_dir),
        "--export_path={}".format(model_path),
        "--use_tpu={}".format("true" if FLAGS.tpu_name else "false"),
        "--tpu_name={}".format(FLAGS.tpu_name),
        "--num_examples={}".format(num_records),
        "--freeze=true",
        *record_paths)

    // Append the time elapsed from when the RL was started to when this model

After Change

    model_path = os.path.join(FLAGS.model_dir, state.train_model_name)

    await checked_run([
        "python3", "train.py",
        "--flagfile={}".format(os.path.join(FLAGS.flags_dir, "train.flags")),
        "--work_dir={}".format(FLAGS.work_dir),
        "--export_path={}".format(model_path),
        "--use_tpu={}".format("true" if FLAGS.tpu_name else "false"),
        "--tpu_name={}".format(FLAGS.tpu_name),
        "--num_examples={}".format(num_records),
        "--freeze=true"] + record_paths)

    // Append the time elapsed from when the RL was started to when this model
    // was trained.

Instances