Commit dc367bae authored by manuschn's avatar manuschn

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])
self.register_variables(self._model.variables)
self._model.summary()
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