(input_blocks, path)
| 114 | |
| 115 | |
| 116 | def save_unet_input_blocks(input_blocks, path): |
| 117 | save_conv2d(input_blocks[0][0], pathlib.Path(path, 'conv')) |
| 118 | save_res_transformer(input_blocks[1], pathlib.Path(path, 'rt1')) |
| 119 | save_res_transformer(input_blocks[2], pathlib.Path(path, 'rt2')) |
| 120 | save_downsample(input_blocks[3][0], pathlib.Path(path, 'd1')) |
| 121 | save_res_transformer(input_blocks[4], pathlib.Path(path, 'rt3')) |
| 122 | save_res_transformer(input_blocks[5], pathlib.Path(path, 'rt4')) |
| 123 | save_downsample(input_blocks[6][0], pathlib.Path(path, 'd2')) |
| 124 | save_res_transformer(input_blocks[7], pathlib.Path(path, 'rt5')) |
| 125 | save_res_transformer(input_blocks[8], pathlib.Path(path, 'rt6')) |
| 126 | save_downsample(input_blocks[9][0], pathlib.Path(path, 'd3')) |
| 127 | save_res_block(input_blocks[10][0], pathlib.Path(path, 'r1')) |
| 128 | save_res_block(input_blocks[11][0], pathlib.Path(path, 'r2')) |
| 129 | |
| 130 | def save_unet_output_blocks(output_blocks, path): |
| 131 | save_res_block(output_blocks[0][0], pathlib.Path(path, 'r1')) |
no test coverage detected