def _init_losses(self): data_type = torch.cuda.FloatTensor self.l1_loss = nn.L1Loss().type(data_type) self.extended_l1_loss = ExtendedL1Loss().type(data_type) self.blur_function = StdLoss().type(data_type) self.gradient_loss = GradientLoss().type(data_type) self.gray_loss = GrayLoss().type(data_type)
def _init_loss(self): data_type = torch.cuda.FloatTensor self.mse_loss = torch.nn.MSELoss().type(data_type) self.blur_loss = StdLoss().type(data_type)
def _init_loss(self): self.mse_loss = torch.nn.MSELoss().type(self.data_type) self.blur_loss = StdLoss().type(self.data_type)