Skip to content

GitLab

Commit dc367bae authored by manuschn's avatar manuschn
Browse files

removed printing model summary from density obs model

parent a9abd557
Hide whitespace changes
Inline Side-by-side
models/global_dens_obs_model.py
View file @ dc367bae
...@@ -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