mirror of
https://github.com/NousResearch/atropos.git
synced 2026-04-24 17:04:55 +00:00
17 lines
619 B
Python
17 lines
619 B
Python
from pathlib import Path
|
|
import argparse
|
|
from pytorch_fob.engine.engine import Engine
|
|
|
|
|
|
if __name__ == "__main__":
|
|
|
|
# parsing
|
|
parser = argparse.ArgumentParser(description="Create a heatmap plot of benchmarking results.")
|
|
parser.add_argument("settings", type=Path,
|
|
help="Path to the experiment yaml file.")
|
|
args, extra_args = parser.parse_known_args()
|
|
if not any(arg.startswith("engine.plot") for arg in extra_args):
|
|
extra_args += ["engine.plot=true"]
|
|
engine = Engine()
|
|
engine.parse_experiment_from_file(args.settings, extra_args=extra_args)
|
|
engine.plot()
|