 - Universal Planning Networks (UPN) [1]}. Aiming to learn a plannable representation using supervision,
UPN assumes a continuous and differentiable action space to enable gradient-based planning by minimizing the supervised imitation loss.
We adapt UPN by adding a softmax layer to output discrete actions.
