Skip to content
Snippets Groups Projects
Commit d7116563 authored by Daniel CARRON's avatar Daniel CARRON :b: Committed by André Anjos
Browse files

[normalizer] Remove target integer check

parent 9a8a4c90
No related branches found
No related tags found
1 merge request!46Create common library
...@@ -35,17 +35,6 @@ def make_z_normalizer( ...@@ -35,17 +35,6 @@ def make_z_normalizer(
# Peek the number of channels of batches in the data loader # Peek the number of channels of batches in the data loader
batch = next(iter(dataloader)) 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] channels = batch[0].shape[1]
# Initialises accumulators # Initialises accumulators
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment