4bd09281f05185e5625112941b7ed11cf4e6cad5,catalyst/rl/agent/critic.py,StateActionCritic,get_from_params,#Any#Any#Any#Any#,138

Before Change

        state_action_net_params["observation_net_params"]["hiddens"]\
            .insert(0, observation_size)

        action_size = reduce(lambda x, y: x * y, env_spec.action_space.shape)
        state_action_net_params["action_net_params"]["hiddens"] \
            .insert(0, action_size)

        value_head_params["out_features"] = 1

        // @TODO: make by init?

After Change

        // @TODO: any better solution?
        state_action_net_params["state_shape"] = \
            env_spec.state_space.shape
        state_action_net_params["action_shape"] = \
            env_spec.action_space.shape
        state_action_net = StateActionNet.get_from_params(
            **state_action_net_params
        )

Instances