\begin{table}[htbp]
\centering
\caption{\textbf{Detailed model configuration for NeuroSketch-Base and NeuroSketch-Large.}}
\label{tab:Table.6}
\begin{tabular}{lcc}
\toprule
                                    & NeuroSketch-Base          & NeuroSketch-Large     \\ \midrule
\# Params (M)                            & 1.4                    & 4.2      \\
$D_\text{stage 1}$                  & 96                        & 96       \\
$D_\text{stage 2}$                  & 128                       & 144              \\
$D_\text{stage 3}$                  & 160                       & 256              \\
$D_\text{stage 4}$                  & 192                       & 384              \\
$d$                                 & 2                         & 3         \\
$G$                                 & 4                         & 4         \\
\bottomrule
\end{tabular}
\end{table}