diff --git a/src/mednet/libs/common/models/normalizer.py b/src/mednet/libs/common/models/normalizer.py index 2eccf0511239d1557557ce6ad16b888291040f2a..cc15adecdeb1176e0d89d35c0b3f0abb16e961e0 100644 --- a/src/mednet/libs/common/models/normalizer.py +++ b/src/mednet/libs/common/models/normalizer.py @@ -35,17 +35,6 @@ def make_z_normalizer( # Peek the number of channels of batches in the data loader batch = next(iter(dataloader)) - - # Ensure targets are ints - try: - target = batch[1]["target"][0].item() - if not isinstance(target, int): - logger.info("Targets are not Integer type, skipping z-normalization.") - return None - except RuntimeError: - logger.info("Targets are not Integer type, skipping z-normalization.") - return None - channels = batch[0].shape[1] # Initialises accumulators