Bugfix SDE val loss, remove early stopping

JonathanCrabbe · Dec 28, 2023 · e8d59cd · e8d59cd
1 parent 00d848b
commit e8d59cd
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 deletions.
diff --git a/cmd/conf/trainer/callbacks/default.yaml b/cmd/conf/trainer/callbacks/default.yaml
@@ -3,9 +3,6 @@
   monitor: val/loss
   filename: "epoch={epoch}-val_loss={val/loss:.2f}"
   auto_insert_metric_name: false
-- _target_: pytorch_lightning.callbacks.EarlyStopping
-  monitor: val/loss
-  patience: 20
 - _target_: fdiff.utils.callbacks.SamplingCallback
   every_n_epochs: 10
   sample_batch_size: ${datamodule.batch_size}

diff --git a/src/fdiff/dataloaders/datamodules.py b/src/fdiff/dataloaders/datamodules.py
@@ -40,6 +40,8 @@ def __init__(
         self.standardize = standardize
         if X_ref is None:
             X_ref = X
+        elif fourier_transform:
+            X_ref = dft(X_ref).detach()
         self.feature_mean = X_ref.mean(dim=0)
         self.feature_std = X_ref.std(dim=0)