python test_model.py \
        problem.hyp.alpha=0.0 \
        problem/model=dt_net_2d_custom \
        problem=sudoku \
        name=sudoku_ablation \
        problem.model.num_class=10 \
        problem.model.model=dt_net_recall_2d \
        problem.model.width=128 \
        problem.hyp.lr=0.001 \
        problem.model.model_path="/home/fis/workspace_AI/AI-RnD/hieutb2/deep-thinking/outputs/sudoku_ablation/training-mundane-Diante/"