Method register_parameter

cdslib/core/models/base_model.py:59–62

(self, name: str, param: torch.Tensor)

            self.register_var_to_save(var_name=name)

    def register_parameter(self, name: str, param: torch.Tensor):
        self.parameter_names.add(name)
        setattr(self, name, param)
        self.register_var_to_load(var_name=name)

    def register_var_to_save(
        self,