Seich
/
mayavoz
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

rmv augmentations

This commit is contained in:
shahules786 2022-10-26 21:47:29 +05:30
parent c51dea6885
commit 47bbee2c32
1 changed files with 8 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
enhancer/cli/train.py
View File

@ -11,7 +11,8 @@ from pytorch_lightning.callbacks import (
)
from pytorch_lightning.loggers import MLFlowLogger
from torch.optim.lr_scheduler import ReduceLROnPlateau
from torch_audiomentations import Compose, Shift
# from torch_audiomentations import Compose, Shift
os.environ["HYDRA_FULL_ERROR"] = "1"
JOB_ID = os.environ.get("SLURM_JOBID", "0")
@ -30,13 +31,13 @@ def main(config: DictConfig):
)
parameters = config.hyperparameters
apply_augmentations = Compose(
[
Shift(min_shift=0.5, max_shift=1.0, shift_unit="seconds", p=0.5),
]
)
# apply_augmentations = Compose(
# [
# Shift(min_shift=0.5, max_shift=1.0, shift_unit="seconds", p=0.5),
# ]
# )
dataset = instantiate(config.dataset, augmentations=apply_augmentations)
dataset = instantiate(config.dataset, augmentations=None)
model = instantiate(
config.model,
dataset=dataset,