xnn.common.finetune.heads.find_multihead#

xnn.common.finetune.heads.find_multihead(model)[source]#

The MultiHead inside a (possibly wrapped) model, if any.

Parameters:

model (Module)

Return type:

MultiHead | None