Method update

shap_e/models/nn/meta.py:203–218

Updates the parameter list before the forward prop so that if `params` is None or doesn't have a certain key, the module uses the default parameter/buffer registered in the module.

(self, params=None)

        return dict(gen)

    def update(self, params=None):
        """
        Updates the parameter list before the forward prop so that if `params`
        is None or doesn't have a certain key, the module uses the default
        parameter/buffer registered in the module.
        """
        if params is None:
            params = AttrDict()
        params = AttrDict(params)
        named_params = set([name for name, _ in self.named_parameters()])
        for name, param in self.named_parameters():
            params.setdefault(name, param)
        for name, param in self.state_dict().items():
            if name not in named_params:
                params.setdefault(name, param)
        return params


def batch_meta_parameters(net, batch_size):