hiro/ray
1
0
Fork 0
mirror of https://github.com/vale981/ray synced 2025-03-06 02:21:39 -05:00
Code Issues Projects Releases Packages Wiki Activity Actions

[RLlib] Fixing conv filters config for ComplexInputNetwork (#14749)

Browse source
This commit is contained in:
Jack Parsons 2021-03-24 15:15:36 +00:00 • committed by GitHub
parent 8874ccec2d
commit 3df7a010b1
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 6 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

5
rllib/models/tf/complex_input_net.py
View file

@ -49,8 +49,9 @@ class ComplexInputNetwork(TFModelV2):
# Image space. # Image space.
if len(component.shape) == 3: if len(component.shape) == 3:
config = { config = {
"conv_filters": model_config.get( "conv_filters": model_config["conv_filters"]
"conv_filters", get_filter_config(component.shape)), if "conv_filters" in model_config else
get_filter_config(obs_space.shape),
"conv_activation": model_config.get("conv_activation"), "conv_activation": model_config.get("conv_activation"),
"post_fcnet_hiddens": [], "post_fcnet_hiddens": [],
} }

5
rllib/models/torch/complex_input_net.py
View file

@ -55,8 +55,9 @@ class ComplexInputNetwork(TorchModelV2, nn.Module):
# Image space. # Image space.
if len(component.shape) == 3: if len(component.shape) == 3:
config = { config = {
"conv_filters": model_config.get( "conv_filters": model_config["conv_filters"]
"conv_filters", get_filter_config(component.shape)), if "conv_filters" in model_config else
get_filter_config(obs_space.shape),
"conv_activation": model_config.get("conv_activation"), "conv_activation": model_config.get("conv_activation"),
"post_fcnet_hiddens": [], "post_fcnet_hiddens": [],
} }