ksuit.utils.model.helpers

Functions

compute_model_norm(module)

Compute the norm of the weights of a nn.Module, useful to check if weight initialization has been correctly

Module Contents

ksuit.utils.model.helpers.compute_model_norm(module)

Compute the norm of the weights of a nn.Module, useful to check if weight initialization has been correctly applied (i.e., the model weights has changed)

Parameters:

module (torch.nn.Module) – nn.Module to compute the norm for.