Skip to content

GitLab

Commit 45fd9249 authored by Abdullah Rashwan's avatar Abdullah Rashwan Committed by A. Unique TensorFlower
Browse files

Internal change 

PiperOrigin-RevId: 397179381
parent 1c79ece9
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 1 deletion
official/vision/beta/projects/simclr/modeling/simclr_model.py
View file @ 45fd9249
...@@ -86,7 +86,8 @@ class SimCLRModel(tf.keras.Model): ...@@ -86,7 +86,8 @@ class SimCLRModel(tf.keras.Model):
features = inputs features = inputs
# Base network forward pass. # Base network forward pass.
endpoints = self._backbone(features, training=training) endpoints = self._backbone(
features, training=training and self._backbone_trainable)
features = endpoints[max(endpoints.keys())] features = endpoints[max(endpoints.keys())]
projection_inputs = layers.GlobalAveragePooling2D()(features) projection_inputs = layers.GlobalAveragePooling2D()(features)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment