Total train-rollout mismatch |log πθ − log μk| 0.045 0.050 0.055 0.060 0.065 0.070 0.075 0 48 96 144 192 240 Optimizer steps Nats