kevslinger / DTQN

Deep Transformer Q-Networks for Partially Observable Reinforcement Learning
MIT License
136 stars 21 forks source link

shrinkwrapping #1

Closed lyu-xg closed 1 year ago

kevslinger commented 1 year ago

Thanks for the PR! I like a lot of what is going on here. I'm going to merge these changes in, and then re-add some of the logging/checkpointing/evaluation code. Cheers!