(args, step_num)
| 122 | |
| 123 | |
| 124 | def get_script(args, step_num): |
| 125 | model_size = get_model_size(args, step_num) |
| 126 | script = os.path.join( |
| 127 | os.getcwd(), |
| 128 | step_dirs[step_num], |
| 129 | "training_scripts", |
| 130 | args.deployment_type, |
| 131 | f"run_{model_size}.sh", |
| 132 | ) |
| 133 | assert os.path.isfile( |
| 134 | script |
| 135 | ), f"{script} does not exist.\n\n Use examples in {os.path.dirname(script)} as a template." |
| 136 | |
| 137 | return script |
| 138 | |
| 139 | |
| 140 | def verify_model(args, step_num): |
no test coverage detected