Method register_parameter

python/oneflow/nn/graph/proxy.py:326–334

(self, name: str, param: Optional[Parameter])

            self.__setattr__(name, module)

    def register_parameter(self, name: str, param: Optional[Parameter]) -> None:
        self.__setattr__(
            name,
            get_proxy_cls(param)(
                param,
                self.to(GraphModule)._name_prefix + self.to(GraphModule)._name + ".",
                name,
            ),
        )

    def modules(self, memo: Optional[Set["Proxy"]] = None) -> Iterator["Proxy"]:
        assert self.to(GraphModule)._type == GraphBlockType.MODULE