c74dc58f8b38bb947b903d3f92efbd8d1cd675ce,rllib/models/torch/fcnet.py,FullyConnectedNetwork,__init__,#FullyConnectedNetwork#Any#Any#Any#Any#Any#,19

Before Change

        // TODO(sven): implement case: vf_shared_layers = False.
        // vf_share_layers = model_config.get("vf_share_layers")

        logger.debug("Constructing fcnet {} {}".format(hiddens, activation))
        layers = []
        prev_layer_size = int(np.product(obs_space.shape))
        self._logits = None

After Change

        self._value_branch_separate = None
        if not self.vf_share_layers:
            // Build a parallel set of hidden layers for the value net.
            prev_vf_layer_size = int(np.product(obs_space.shape))
            self._value_branch_separate = []
            for size in hiddens:
                self._value_branch_separate.append(
                    SlimFC(
                        in_size=prev_vf_layer_size,
                        out_size=size,
                        activation_fn=activation,
                        initializer=normc_initializer(1.0)))
                prev_vf_layer_size = size
            self._value_branch_separate = nn.Sequential(
                *self._value_branch_separate)

Instances