(self)
| 166 | return res |
| 167 | |
| 168 | def fuse_model(self): |
| 169 | for m in self.modules(): |
| 170 | if type(m) == ConvRelu: |
| 171 | # print("fuse conv and relu in ConvRelu") |
| 172 | torch.quantization.fuse_modules(m, ['conv', 'relu'], inplace=True) |
| 173 | if type(m) == Gblock: |
| 174 | # print("fuse conv and relu in Gblock") |
| 175 | torch.quantization.fuse_modules(m, ['conv0', 'relu'], inplace=True) |
| 176 | |
| 177 | |
| 178 | if __name__ == '__main__': |