def forward(self, dehaze, gt):
    loss = []
    dehaze_features = self.output_features(dehaze)
    gt_features = self.output_features(gt)
    for (dehaze_feature, gt_feature) in zip(dehaze_features, gt_features):
        loss.append(F.mse_loss(dehaze_feature, gt_feature))
    return (sum(loss) / len(loss))
def forward(self, dehaze, gt):
    loss = []
    dehaze_features = self.output_features(dehaze)
    gt_features = self.output_features(gt)
    for (dehaze_feature, gt_feature) in zip(dehaze_features, gt_features):
        loss.append(F.mse_loss(dehaze_feature, gt_feature))
    return (sum(loss) / len(loss))