cellflow.networks.TokenAttentionPooling.num_heads

TokenAttentionPooling.num_heads: int = 8