Please run these commands from the root directory of the project.
# Train
$ jac-run scripts/blocksworld/learn_policy.py --task final
# Test
$ jac-run scripts/blocksworld/learn_policy.py --task final --test-only --load CHECKPOINT
# add [--test-epoch-size T] to control the number of testing cases.