(self,x)
| 61 | torch.nn.Tanh() |
| 62 | ) |
| 63 | def forward(self,x): |
| 64 | x = self.initial(x) |
| 65 | for layer in self.down_blocks: |
| 66 | x = layer(x) |
| 67 | x = self.residual_blocks(x) |
| 68 | for layer in self.up_blocks: |
| 69 | x = layer(x) |
| 70 | out = self.last(x) |
| 71 | return out |
| 72 | |
| 73 | if __name__ == '__main__': |
| 74 | img_channels = 3 |
nothing calls this directly
no outgoing calls
no test coverage detected