(args, step_num)
| 104 | |
| 105 | |
| 106 | def get_model_size(args, step_num): |
| 107 | if step_num == 3: |
| 108 | return get_model_size(args, 1) |
| 109 | return getattr(args, f"{model_type[step_num]}_model") |
| 110 | |
| 111 | |
| 112 | def get_zero_stage(args, step_num): |
no outgoing calls
no test coverage detected