check_csv() — pytorch Function Reference

Architecture documentation for the check_csv() function in check_csv.py from the pytorch codebase.

Function python DynamoBenchmarks Microbenchmarking called by 1

Entity Profile

DynamoBenchmarks→ Microbenchmarking→ check_csv() — pytorch Function Reference

Dependency Diagram

graph TD
  7bb33263_0887_aa80_c6db_42a0d46d8c2f["check_csv()"]
  4209e548_c6cc_7d8b_5cc7_4e69800314cb["parser()"]
  4209e548_c6cc_7d8b_5cc7_4e69800314cb -->|calls| 7bb33263_0887_aa80_c6db_42a0d46d8c2f
  style 7bb33263_0887_aa80_c6db_42a0d46d8c2f fill:#6366f1,stroke:#818cf8,color:#fff

Relationship Graph

Source Code

benchmarks/dynamo/check_csv.py lines 8–33

def check_csv(filename):
    """
    Basic accuracy checking.
    """

    df = pd.read_csv(filename)

    failed = []
    for _, row in df.iterrows():
        model_name = row["name"]
        status = row["accuracy"]
        if "pass" not in status:
            failed.append(model_name)

        print(f"{model_name:34} {status}")

    if failed:
        print(
            textwrap.dedent(
                f"""
                Error {len(failed)} models failed
                    {" ".join(failed)}
                """
            )
        )
        sys.exit(1)