(self, use_gpu=True, colorspace='Lab')
| 143 | # L2, DSSIM metrics |
| 144 | class FakeNet(nn.Module): |
| 145 | def __init__(self, use_gpu=True, colorspace='Lab'): |
| 146 | super(FakeNet, self).__init__() |
| 147 | self.use_gpu = use_gpu |
| 148 | self.colorspace=colorspace |
| 149 | |
| 150 | class L2(FakeNet): |
| 151 |