<cite>Tu and Gimpel (2018)</cite> used the same inference network for solving both problems, which led <cite>them</cite> to fine-tune the network at test-time with a different objective. We avoid this issue by training two inference networks, A Ψ for test-time inference and F Φ for cost-augmented inference: