Python main Beispiele

Beispiel #1

 def run_benchmark_gpt2(self, arguments: str):
     from benchmark_gpt2 import parse_arguments, main
     args = parse_arguments(arguments.split())
     csv_filename = main(args)
     self.assertTrue(os.path.exists(csv_filename))

Beispiel #2

Datei: test_gpt2.py Projekt: xadupre/onnxruntime

 def run_benchmark_gpt2(self, arguments: str):
     args = parse_arguments(arguments.split())
     csv_filename = main(args)
     self.assertIsNotNone(csv_filename)
     self.assertTrue(os.path.exists(csv_filename))