fix

model.py

@@ -34,7 +34,7 @@ def build_classifier_head(input_size, config: Config):
   # inputs = keras.Input(shape=(input_size,))
   # classifier = AttentionDenseClassifierHead(2, config.num_classes)(inputs)
   # model = keras.Model(inputs=inputs, outputs=classifier)
-  return keras.Layers.Dense(config.num_classes, activation='softmax')
+  return keras.layers.Dense(config.num_classes, activation='softmax')
 
 def build_model(movinet, classifier_head):
   return keras.models.Sequential([movinet, classifier_head])