(self, x: torch.Tensor)
| 59 | self.convtranspose3.bias.copy_(torch.rand_like(self.convtranspose3.bias)) |
| 60 | |
| 61 | def forward(self, x: torch.Tensor) -> torch.Tensor: |
| 62 | x = self.conv1(x) |
| 63 | x = self.conv2(x) |
| 64 | x = self.conv3(x) |
| 65 | x = self.convtranspose1(x) |
| 66 | x = self.convtranspose2(x) |
| 67 | x = self.convtranspose3(x) |
| 68 | return x |
| 69 | |
| 70 | model = MyModel().cuda() |
| 71 | dump_torch_to_onnx(model=model, onnx_export_file='model.onnx', input_shape=[1, 8, 96, 96]) |
no outgoing calls
no test coverage detected