(model, pagelocked_buffer)
| 119 | |
| 120 | # Loads a random test case from pytorch's DataLoader |
| 121 | def load_random_test_case(model, pagelocked_buffer): |
| 122 | # Select an image at random to be the test case. |
| 123 | img, expected_output = model.get_random_testcase() |
| 124 | # Copy to the pagelocked input buffer |
| 125 | np.copyto(pagelocked_buffer, img) |
| 126 | return expected_output |
| 127 | |
| 128 | |
| 129 | def main(): |
no test coverage detected