2 ms·

Thanks! We still haven’t figured it a way to get this method to work directly for high dimensional obs space (just used a VAE to compress it for now)