nemo_automodel.components.checkpoint.utils#
Module Contents#
Functions#
Check if the model’s word embeddings are tied. |
API#
- nemo_automodel.components.checkpoint.utils.is_tied_word_embeddings(model: torch.nn.Module) bool#
Check if the model’s word embeddings are tied.
- Parameters:
model (nn.Module) – The model to check.
- Returns:
True if the model’s word embeddings are tied, False otherwise.
- Return type:
bool