(self,x)
| 7 | super(MyModule,self).__init__() |
| 8 | self.conv1 = nn.Conv2d(1,3,3) |
| 9 | def forward(self,x): |
| 10 | x = self.conv1(x) |
| 11 | return x |
| 12 | |
| 13 | model = MyModule() # 实例化模型 |
| 14 | trace_module = torch.jit.trace(model,torch.rand(1,1,224,224)) |
nothing calls this directly
no outgoing calls
no test coverage detected