SlotAttention.ConfigΒΆ

Component: SlotAttention

class SlotAttention.Config[source]

Bases: ConfigBase

All Attributes (including base classes)

attn_dimension: int = 64
attention_type: SlotAttentionType = <SlotAttentionType.NO_ATTENTION: 'no_attention'>

Default JSON

{
    "attn_dimension": 64,
    "attention_type": "no_attention"
}