search

lifelong-learning-systems / rlblocks

Reinforcement Learning Blocks for Researchers
MIT License
0 stars 0 forks source link

Draft: datasets & samplers api #2

Closed coreylowman closed 2 years ago

coreylowman commented 2 years ago

I think there's a missing piece to this code, which is normally called collation in pytorch. that is transforming a list of transitions in a batch. I think we can probably do that in the batch sampler classes? Edit: i added a collate function

coreylowman commented 2 years ago

Need to think about how to put generative memory fits in here