Forward pass of the model Args: obs_img (torch.Tensor): batch of observations goal_img (torch.Tensor): batch of goals Returns: dist_pred (torch.Tensor): predicted distance to goal action_pred (torch.Tensor): predicted action
(
self, obs_img: torch.tensor, goal_img: torch.tensor
)
source not stored for this graph (policy: none)
nothing calls this directly
no outgoing calls
no test coverage detected