mirror of
https://github.com/NousResearch/atropos.git
synced 2026-04-24 17:04:55 +00:00
22 lines
613 B
Python
22 lines
613 B
Python
import argparse
|
|
from pathlib import Path
|
|
from pytorch_fob.engine.engine import Engine
|
|
|
|
|
|
def get_parser():
|
|
parser = argparse.ArgumentParser()
|
|
parser.add_argument("experiment_file", type=Path,
|
|
help="The yaml file specifying the experiment.")
|
|
return parser
|
|
|
|
|
|
def main(args: argparse.Namespace, extra_args: list[str]):
|
|
engine = Engine()
|
|
engine.parse_experiment_from_file(args.experiment_file, extra_args=extra_args)
|
|
engine.prepare_data()
|
|
|
|
|
|
if __name__ == '__main__':
|
|
parser = get_parser()
|
|
args, extra_args = parser.parse_known_args()
|
|
main(args, extra_args)
|