\begin{figure}[t!]
    \centering
    \includegraphics[width=\linewidth]{figure/source/experiments/mask_probs.png}
    \caption{{Prediction of the \csi{} for the sub-actions corresponding to the key ($\gA_{\text{pick}}$) and the door ($\gA_{\text{open}}$) along the trajectory in DoorKey.}}
    \label{fig:mask_probs}
    % \vspace{-10pt}
\end{figure}
