Commit dc367bae authored by manuschn's avatar manuschn
Browse files

removed printing model summary from density obs model

parent a9abd557
......@@ -34,7 +34,6 @@ class GlobalDensObsModel(TFModelV2):
baseline = tf.keras.layers.Dense(units=1)(conv_out)
self._model = tf.keras.Model(inputs=observations, outputs=[logits, baseline])
def forward(self, input_dict, state, seq_lens):
if self._mask_unavailable_actions:
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