Some questions about the code

Hello, I am currently doing some related research based on this project, but I can't understand lines 158 to 162 in CURE/src/trainer/gpt_conut_trainer.py, so I want to know what and how they work? This will be very helpful for me, thank you very much