Skip to content

Commit 783368e

Browse files
Update cifar10-example.py
AI suggested tf 2.15.0 -> 2.19.0 compat fix.
1 parent 4806719 commit 783368e

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

cifar10-example.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -88,8 +88,8 @@ def make_dataset(dataset):
8888

8989
last_relevant_layer = base_new.layers[-2]
9090
# last_relevant_layer_extracted = last_relevant_layer #.output[0][0][0]
91-
base_embedding = tf.keras.Model(inputs=base_new.layers[0].input,
92-
outputs=last_relevant_layer.output)
91+
base_embedding = tf.keras.Model(inputs=base_new.input,
92+
outputs=last_relevant_layer.output)
9393

9494

9595
image_input_0 = tf.keras.layers.Input(shape=INPUT_SHAPES[0])

0 commit comments

Comments
 (0)