def include(self, params):
        """
        Include the gradients for a set of parameters, normally the final embedding layer.

        Args:
            params: PyTorch Network layer after .backward() was called.
        Returns:
            Nothing!
        """
        gradients = [params.weight.grad.detach().cpu().numpy()]

        for grad in gradients:
            ### Shape: 128 x 2048
            self.saver["grad_normal_mean"].append(np.mean(grad, axis=0))
            self.saver["grad_normal_std"].append(np.std(grad, axis=0))
            self.saver["grad_abs_mean"].append(np.mean(np.abs(grad), axis=0))
            self.saver["grad_abs_std"].append(np.std(np.abs(grad), axis=0))

        Include the gradients for a set of parameters, normally the final embedding layer.

        Args:
            params: PyTorch Network layer after .backward() was called.
        Returns:
            Nothing!
        
examples/pytorch/hilander/PSS/Smooth_AP/src/evaluate.py