(conv2d, path)
| 50 | return Tensor([val, val]) |
| 51 | |
| 52 | def save_conv2d(conv2d, path): |
| 53 | pathlib.Path(path).mkdir(parents=True, exist_ok=True) |
| 54 | |
| 55 | save_tensor(conv2d.weight, 'weight', path) |
| 56 | if conv2d.bias is not None: |
| 57 | save_tensor(conv2d.bias, 'bias', path) |
| 58 | save_tensor(to_tuple_tensor(conv2d.stride), 'stride', path) |
| 59 | save_tensor(to_tuple_tensor(conv2d.padding), 'padding', path) |
| 60 | save_tensor(to_tuple_tensor(conv2d.dilation), 'dilation', path) |
| 61 | save_scalar(conv2d.groups, "n_group", path) |
| 62 | save_tensor(to_tuple_tensor(conv2d.kernel_size), 'kernel_size', path) |
| 63 | |
| 64 | assert conv2d.groups == 1 |
| 65 | in_channels = conv2d.weight.shape[1] |
| 66 | out_channels = conv2d.weight.shape[0] |
| 67 | save_scalar(in_channels, "n_channels_in", path) |
| 68 | save_scalar(out_channels, "n_channels_out", path) |
| 69 | |
| 70 | def save_padded_conv2d(padded_conv2d, path): |
| 71 | pathlib.Path(path).mkdir(parents=True, exist_ok=True) |
no test coverage detected