FlattenConcatBaseNet¶
-
class
maze.perception.models.built_in.flatten_concat.
FlattenConcatBaseNet
(*args: Any, **kwargs: Any)¶ Base flatten and concatenation model for policies and critics.
- Parameters
obs_shapes – Dictionary mapping of observation names to shapes.
hidden_units – Dictionary mapping of action names to shapes.
non_lin – The non-linearity to apply.