(self)
| 47 | |
| 48 | class BaseNet(nn.Module): |
| 49 | def __init__(self): |
| 50 | super(BaseNet, self).__init__() |
| 51 | |
| 52 | # register buffer |
| 53 | self.register_buffer( |
| 54 | 'mean', torch.Tensor([-.030, -.088, -.188])[None, :, None, None]) |
| 55 | self.register_buffer( |
| 56 | 'std', torch.Tensor([.458, .448, .450])[None, :, None, None]) |
| 57 | |
| 58 | def set_requires_grad(self, state: bool): |
| 59 | for param in chain(self.parameters(), self.buffers()): |