From d71165634db771c3fd1453e76d4847de4a2efccd Mon Sep 17 00:00:00 2001
From: dcarron <daniel.carron@idiap.ch>
Date: Tue, 25 Jun 2024 18:05:09 +0200
Subject: [PATCH] [normalizer] Remove target integer check
---
src/mednet/libs/common/models/normalizer.py | 11 -----------
1 file changed, 11 deletions(-)
diff --git a/src/mednet/libs/common/models/normalizer.py b/src/mednet/libs/common/models/normalizer.py
index 2eccf051..cc15adec 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
--
GitLab