mattricesound commited on
Commit
1530829
1 Parent(s): 7d6db8f

Remove random seed in datasets as it is created from pl

Browse files
Files changed (2) hide show
  1. remfx/datasets.py +0 -1
  2. scripts/train.py +2 -2
remfx/datasets.py CHANGED
@@ -11,7 +11,6 @@ from typing import Any, List, Tuple
11
 
12
  LENGTH = 2**18 # 12 seconds
13
  ORIG_SR = 48000
14
- torch.manual_seed(123)
15
 
16
 
17
  class GuitarFXDataset(Dataset):
 
11
 
12
  LENGTH = 2**18 # 12 seconds
13
  ORIG_SR = 48000
 
14
 
15
 
16
  class GuitarFXDataset(Dataset):
scripts/train.py CHANGED
@@ -9,8 +9,8 @@ log = utils.get_logger(__name__)
9
  @hydra.main(version_base=None, config_path="../", config_name="config.yaml")
10
  def main(cfg: DictConfig):
11
  # Apply seed for reproducibility
12
- print(cfg)
13
- pl.seed_everything(cfg.seed)
14
 
15
  log.info(f"Instantiating datamodule <{cfg.datamodule._target_}>.")
16
  datamodule = hydra.utils.instantiate(cfg.datamodule, _convert_="partial")
 
9
  @hydra.main(version_base=None, config_path="../", config_name="config.yaml")
10
  def main(cfg: DictConfig):
11
  # Apply seed for reproducibility
12
+ if cfg.seed:
13
+ pl.seed_everything(cfg.seed)
14
 
15
  log.info(f"Instantiating datamodule <{cfg.datamodule._target_}>.")
16
  datamodule = hydra.utils.instantiate(cfg.datamodule, _convert_="partial")