Function torch_dfs

models/mutual_self_attention.py:12–16

(model: torch.nn.Module)

def torch_dfs(model: torch.nn.Module):
    result = [model]
    for child in model.children():
        result += torch_dfs(child)
    return result


class ReferenceAttentionControl: