python test_per_sample.py \
        problem=mnist \
        name=mnist_dt_net_attention_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-17 14:21:55.707067"