lowrollr / turbozero

fast + parallel AlphaZero in JAX
Apache License 2.0
79 stars 6 forks source link

allow for user-specified data augmentation in Trainer #10

Closed lowrollr closed 5 months ago

lowrollr commented 7 months ago

It might be a good idea for me to allow for users to specify any number of transforms to apply to augment experiences prior to storing in replay memory -- this is a fairly common use-case so it ideally should not require a custom class.

Originally posted by @lowrollr in https://github.com/lowrollr/turbozero/issues/8#issuecomment-1925047026

lowrollr commented 5 months ago

15