Skip to content

allenact.embodiedai.models.aux_models#

[view_source]

Several of the models defined in this file are modified versions of those found in https://github.com/joel99/habitat-pointnav-aux/blob/master/habitat_baselines/

AuxiliaryModel#

class AuxiliaryModel(nn.Module)

[view_source]

The class of defining the models for all kinds of self-supervised auxiliary tasks