mirror of
https://github.com/NousResearch/atropos.git
synced 2026-04-24 17:04:55 +00:00
Convert FOB submodule to regular folder
This commit is contained in:
parent
94f046ad40
commit
94825011a0
74 changed files with 4563 additions and 0 deletions
|
|
@ -0,0 +1,17 @@
|
|||
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()
|
||||
Loading…
Add table
Add a link
Reference in a new issue