Skip to content

allenact.embodiedai.models.fusion_models#

[view_source]

Fusion#

class Fusion(nn.Module)

[view_source]

Base class of belief fusion model from Auxiliary Tasks Speed Up Learning PointGoal Navigation (Ye, 2020) Child class should implement get_belief_weights function to generate weights to fuse the beliefs from all the auxiliary task into one.

SoftmaxFusion#

class SoftmaxFusion(Fusion)

[view_source]

Situational Fusion of Visual Representation for Visual Navigation https://arxiv.org/abs/1908.09073

AttentiveFusion#

class AttentiveFusion(Fusion)

[view_source]

Attention is All You Need https://arxiv.org/abs/1706.03762 i.e. scaled dot-product attention