()
| 68 | |
| 69 | |
| 70 | def test_convert_to_onnx(): |
| 71 | test_assets.add_folder("test_convert_to_onnx") |
| 72 | |
| 73 | model = sam_32_32() |
| 74 | q_model = quantize_model(model) |
| 75 | # Create experiment specific directory |
| 76 | tf.keras.models.save_model( |
| 77 | q_model, test_assets.test_convert_to_onnx.int8_saved_model |
| 78 | ) |
| 79 | convert_saved_model_to_onnx( |
| 80 | saved_model_dir=test_assets.test_convert_to_onnx.int8_saved_model, |
| 81 | onnx_model_path=test_assets.test_convert_to_onnx.int8_onnx_model, |
| 82 | ) |
| 83 | tf.keras.backend.clear_session() |
| 84 | |
| 85 | |
| 86 | def test_find_my_predecessors(): |
nothing calls this directly
no test coverage detected