Stuck in memory._retrieve when batch size > 32

Hi,

I notice that RAINBOW doesn't work when the batch size is greater than 32 (I tried for 64, 128, 256), where it is stuck in memory._retrieve the recursive call. Why does this happen? Is there something that I can do about this (to increase the batch size) or batch size needs to be small?

Thanks

This is probably an issue with PER sampling. Try the fix in https://github.com/Kaixhin/Rainbow/issues/41 and see if that helps.