Convert FOB submodule to regular folder

This commit is contained in:
arihanv 2025-05-18 16:36:28 -07:00
parent 94f046ad40
commit 94825011a0
74 changed files with 4563 additions and 0 deletions

View file

@ -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()