Method execute

demo/HuggingFace/run.py:127–142

(self, args: argparse.Namespace)

class BenchmarkAction(NetworkScriptAction):
    def execute(self, args: argparse.Namespace):
        module = self.load_script(args.script, args)
        module.RUN_CMD._parser = self.parser

        old_path = os.getcwd()
        # Execute script in each relevant folder
        try:
            os.chdir(args.network)
            results = module.RUN_CMD.run_benchmark()
        finally:
            os.chdir(old_path)

        # Output to terminal
        print(results)

        return 0

    def add_args(self, parser: argparse.ArgumentParser):
        super().add_args(parser)