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)
