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