(model_name)
| 124 | |
| 125 | |
| 126 | def get_sample_inputs(model_name): |
| 127 | # Lock the random seed for reproducibility |
| 128 | torch.manual_seed(42) |
| 129 | |
| 130 | if is_vision_model(model_name): |
| 131 | return get_vision_model_sample_input() |
| 132 | if model_name == "YOLO_NAS_S": |
| 133 | input_batch = get_dog_image_tensor(640) |
| 134 | return input_batch |
| 135 | |
| 136 | return None |
| 137 | |
| 138 | |
| 139 | def get_dynamic_shapes(model_name): |
no test coverage detected