Skip to content

projects.pointnav_baselines.models.point_nav_models#

[view_source]

PointNavActorCritic#

class PointNavActorCritic(VisualNavActorCritic)

[view_source]

Use raw image as observation to the agent

ResnetTensorPointNavActorCritic#

class ResnetTensorPointNavActorCritic(VisualNavActorCritic)

[view_source]

Use resnet_preprocessor to generate observations to the agent

ResnetTensorPointNavActorCritic.is_blind#

 | @property
 | is_blind() -> bool

[view_source]

True if the model is blind (e.g. neither 'depth' or 'rgb' is an input observation type).

ResnetTensorGoalEncoder#

class ResnetTensorGoalEncoder(nn.Module)

[view_source]

ResnetTensorGoalEncoder.get_object_type_encoding#

 | get_object_type_encoding(observations: Dict[str, torch.FloatTensor]) -> torch.FloatTensor

[view_source]

Get the object type encoding from input batched observations.

ResnetDualTensorGoalEncoder#

class ResnetDualTensorGoalEncoder(nn.Module)

[view_source]

ResnetDualTensorGoalEncoder.get_object_type_encoding#

 | get_object_type_encoding(observations: Dict[str, torch.FloatTensor]) -> torch.FloatTensor

[view_source]

Get the object type encoding from input batched observations.