Spaces:
Runtime error
Runtime error
mattricesound
commited on
Commit
•
1530829
1
Parent(s):
7d6db8f
Remove random seed in datasets as it is created from pl
Browse files- remfx/datasets.py +0 -1
- 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 |
-
|
13 |
-
|
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")
|