Hello, my name is Yooseok Lim.
I appreciate your well-structured and qualifying code.
I have a question regarding your implementation. When I followed your code, I observed a very low loss and accuracy in the behavior cloning model. Did you obtain similar results in your experiment, or is this issue specific to my case?

Thank you for your time and consideration.
Best regards,
Yooseok Lim