ce96b03b07a4a4bdd851aa84493c616cd291aff2,rllib/models/torch/torch_modelv2.py,TorchModelV2,variables,#TorchModelV2#Any#,50

Before Change

    @override(ModelV2)
    def variables(self, as_dict: bool = False) -> List[TensorType]:
        if as_dict:
            return self.state_dict()
        return list(self.parameters())

    @override(ModelV2)
    def trainable_variables(self, as_dict: bool = False) -> List[TensorType]:

After Change

    def variables(self, as_dict: bool = False) -> List[TensorType]:
        p = list(self.parameters())
        if as_dict:
            return {k: p[i] for i, k in enumerate(self.state_dict().keys())}
        return p

    @override(ModelV2)
    def trainable_variables(self, as_dict: bool = False) -> List[TensorType]:

Instances