Skip to content

Commit

Permalink
need to normalize images
Browse files Browse the repository at this point in the history
  • Loading branch information
erinmgraham committed Oct 12, 2023
1 parent ded61ea commit 59d83f3
Showing 1 changed file with 4 additions and 0 deletions.
4 changes: 4 additions & 0 deletions episodes/scripts/fit_ep_dropout_model.py
Original file line number Diff line number Diff line change
Expand Up @@ -16,6 +16,10 @@
# load the cifar dataset included with the keras packages
(train_images, train_labels), (val_images, val_labels) = keras.datasets.cifar10.load_data()

# normalize the RGB values to be between 0 and 1
train_images = train_images / 255.0
val_images = val_images / 255.0

# Recall how we compiled our intro and pooling models:

#model_pool.compile(optimizer = 'adam',
Expand Down

0 comments on commit 59d83f3

Please sign in to comment.