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.