620d2c767c309327aa04210192b08f1c1fd0de76,reagent/net_builder/discrete_dqn/dueling.py,Dueling,build_q_network,#Dueling#Any#Any#Any#,26

Before Change

        output_dim: int,
    ) -> ModelBase:
        state_dim = self._get_input_dim(state_normalization_parameters)
        return DuelingQNetwork(
            layers=[state_dim] + self.sizes + [output_dim],
            activations=self.activations + ["linear"],
        )

After Change

        output_dim: int,
    ) -> ModelBase:
        state_dim = self._get_input_dim(state_normalization_parameters)
        return DuelingQNetwork(
            action_dim=output_dim,
            embedding=FullyConnectedNetwork(
                layers=[state_dim] + self.sizes, activations=self.activations
            ),
            quantiles=self.quantiles,
        )

Instances