We adopt the same learning framework as <cite>Tu & Gimpel</cite> of jointly learning the energy function and inference network, but we propose a novel objective function that jointly trains a cost-augmented inference network, a test-time inference network, and the energy function.