(args: argparse.Namespace)
| 136 | |
| 137 | |
| 138 | def get_pte_base_name(args: argparse.Namespace) -> str: |
| 139 | pte_name = args.model_name |
| 140 | if args.compile: |
| 141 | pte_name += "_compiled" |
| 142 | pte_name = f"{pte_name}_coreml_{args.compute_unit}" |
| 143 | return pte_name |
| 144 | |
| 145 | |
| 146 | def save_processed_bytes(processed_bytes, base_name: str): |