Added note about incomplete SDE implementation

This commit is contained in:
Dominik Moritz Roth 2023-09-10 09:35:08 +02:00
parent db8eadbd9d
commit 04e4364482

View File

@ -57,6 +57,7 @@ def get_colored_noise(beta, known_shape=None):
class SDE_Noise(): class SDE_Noise():
def __init__(self, shape, latent_sde_dim=64, Base_Noise=White_Noise): def __init__(self, shape, latent_sde_dim=64, Base_Noise=White_Noise):
raise Exception('Not implemented yet. Just use SB3s gSDE...')
self.shape = shape self.shape = shape
self.latent_sde_dim = latent_sde_dim self.latent_sde_dim = latent_sde_dim
self.Base_Noise = Base_Noise self.Base_Noise = Base_Noise