def _check_one_layer(self, layer, *args, **kwargs):
    self._check_one_layer_with_criterion(layer, nn.L1Loss(reduction='mean'), *args, **kwargs)
    self._check_one_layer_with_criterion(layer, nn.L1Loss(reduction='sum'), *args, **kwargs)
def _check_one_layer(self, layer, *args, **kwargs):
    self._check_one_layer_with_criterion(layer, nn.L1Loss(reduction='mean'), *args, **kwargs)
    self._check_one_layer_with_criterion(layer, nn.L1Loss(reduction='sum'), *args, **kwargs)