Learning to Achieve Goals with Belief State Transformers