Is Imitation All You Need? Generalized Decision-Making with Dual-Phase Training