Curious to know how close that training loop is to actual openai code.