Switch from gym logger to something more capable

This also would probably solve the error of calling the logger after it was already deleted when exiting training due to no control over its lifespan.