Added note about incomplete SDE implementation

2023-09-10 09:35:08 +02:00 · 2023-09-10 09:35:08 +02:00 · 04e4364482
commit 04e4364482
parent db8eadbd9d
1 changed files with 1 additions and 0 deletions
--- a/priorConditionedAnnealing/noise.py
+++ b/priorConditionedAnnealing/noise.py
@ -57,6 +57,7 @@ def get_colored_noise(beta, known_shape=None):

 class SDE_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.latent_sde_dim = latent_sde_dim
        self.Base_Noise = Base_Noise