Method register_parameter

lightx2v/common/modules/weight_module.py:16–18

(self, name, param)

        setattr(self, name, module)

    def register_parameter(self, name, param):
        self._parameters[name] = param
        setattr(self, name, param)

    def load(self, weight_dict):
        for _, module in self._modules.items():