Chore: Deleted not used codeblock

Flippchen · Feb 8, 2024 · df05e45 · df05e45
1 parent 52024b2
commit df05e45
Showing 1 changed file with 0 additions and 5 deletions.
diff --git a/training/train.py b/training/train.py
@@ -53,11 +53,6 @@
 train_ds = train_ds.cache().shuffle(1000).prefetch(buffer_size=AUTOTUNE)
 val_ds = val_ds.cache().prefetch(buffer_size=AUTOTUNE)
 
-# Normalize the data
-normalization_layer = layers.Rescaling(1. / 255)
-normalized_ds = train_ds.map(lambda x, y: (normalization_layer(x), y))
-image_batch, labels_batch = next(iter(normalized_ds))
-
 # Create data augmentation layer and show augmented batch
 data_augmentation = create_augmentation_layer(img_height, img_width)
 show_augmented_batch(train_ds, data_augmentation)