2eb844885d5b810cc93968f4965c58eb96e86d58,prep_model.py,,main,#,28

Before Change

    shutil.copy2(model_file_path, gl_model_file_path)

    pt_dir_path = os.path.join(root_dir_path, "pt-{}".format(model_name))
    os.mkdir(pt_dir_path)
    shutil.copy2(log_file_path, pt_dir_path)
    pt_model_file_path = os.path.join(pt_dir_path, "{}.pth".format(model_name))
    command = "python3 convert_models.py --src-fwk=gluon --dst-fwk=pytorch --src-model={model_name}" \
              " --dst-model={model_name} --src-params={model_file_path} --dst-params={pt_model_file_path}" \

After Change

    top5s.append(top5_err)
    sha1s.append(sha1_value)

    slice_df_dict = {
        "Type": types,
        "Top5": top5s,
        "Sha1": sha1s,
    }
    slice_df = pd.DataFrame(slice_df_dict)
    slice_df.to_csv(
        os.path.join(root_dir_path, "info.csv"),
        sep="\t",

Instances