python test_model.py \
        problem=mnist \
        name=mnist_test_dt_2d \
        problem.model.num_class=10 \
        problem/model=dt_net_2d_custom \
        problem.model.model_path="/home/tranhieu/workdir/research/deep-thinking/outputs/mnist_ablation/training-2024-10-18 16:01:53.630302"