Add fourier flags to train config

Co-authored-by: nicolashuynh <[email protected]>
JonathanCrabbe · Dec 1, 2023 · fcecb26 · fcecb26
1 parent 397582a
commit fcecb26
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/cmd/conf/datamodule/ecg.yaml b/cmd/conf/datamodule/ecg.yaml
@@ -1,4 +1,5 @@
 _target_: fdiff.dataloaders.datamodules.ECGDatamodule
 data_dir: ${hydra:runtime.cwd}/data
 random_seed: ${random_seed}
+fourier_transform: ${fourier_transform}
 batch_size: 64
diff --git a/cmd/conf/train.yaml b/cmd/conf/train.yaml
@@ -1,4 +1,5 @@
 random_seed: 42
+fourier_transform: false
 defaults:
   - _self_
   - score_model: default