Hey There SlimyRat21
We did a small integration of Trains with a Doom agent that uses reinforcement learning.
https://github.com/erezalg/ViZDoom
What we did is basically change a bit the strcuture of how parameters are cought (so we can modify them from the UI), then logged stuff like loss, location on the map, frame buffers at certain times and information about end of episode that might be helpful for us.
You can see how it looks on the demoapp (as long as it lasts 🙂 )
Let me know if you want to consult further!
https://demoapp.trains.allegro.ai/projects/066fc0bddf6147ba8ad9dbf3069e1b1d/experiments/ac85011177d24b09895509d42a8abd7d/execution?columns=selected&columns=type&columns=name&columns=tags&columns=status&columns=project.name&columns=user.name&columns=started&columns=last_update&columns=last_iteration&order=last_update