\begin{table}[]
    \centering
    \begin{tabular}{|c|c|c|c|c|c|c|c|c|}
        \hline
        & epochs & z dim & h dim & RNN dim & adv. dim & disc. cap. & cont. cap. & lr  \\
        \hline
        Synthetic & 50 & 4 & 16 & 16 & 8 & 0.6 & - & 0.0002 \\
        \hline
        CalMS21 & 30 & 8 & 256 & 256 & 8 & 0.69 & 10 & 0.0001  \\
        \hline
        Basketball & 20 & 8 & 128 & 128 & 8 & 0.6 & 4 & 0.02 \\
        \hline
    \end{tabular}
    \caption{Hyperparameters for VAE training. The batch size is the same as the ones for program learning in Table~\ref{tab:hyperparams_prog}.}
    \label{tab:hyperparams_vae}
\end{table}
epochs z dim h dim RNN dim adv. dim disc. cap. cont. cap. lr
Synthetic 50 4 16 16 8 0.6 - 0.0002
CalMS21 30 8 256 256 8 0.69 10 0.0001
Basketball 20 8 128 128 8 0.6 4 0.02