Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in
Toggle navigation
Menu
Open sidebar
Flatland
neurips2020-flatland-baselines
Commits
dc367bae
Commit
dc367bae
authored
Jul 08, 2020
by
manuschn
Browse files
removed printing model summary from density obs model
parent
a9abd557
Changes
1
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
:
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment