Fix self-train RL restoration from another checkpoint.

Fix self-train RL restoration from another checkpoint. #122