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

set sr to dataset sr

This commit is contained in:
shahules786 2022-09-29 17:20:34 +05:30
parent 4e033d2ab5
commit 79525df76e
2 changed files with 11 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
enhancer/models/demucs.py
View File

@ -1,3 +1,4 @@
import logging
from typing import Optional, Union, List from typing import Optional, Union, List
from torch import nn from torch import nn
import torch.nn.functional as F import torch.nn.functional as F
@ -112,6 +113,10 @@ class Demucs(Model):
): ):
duration = dataset.duration if isinstance(dataset,EnhancerDataset) else None duration = dataset.duration if isinstance(dataset,EnhancerDataset) else None
if dataset is not None:
if sampling_rate!=dataset.sampling_rate:
logging.warn(f"model sampling rate {sampling_rate} should match dataset sampling rate {dataset.sampling_rate}")
sampling_rate = dataset.sampling_rate
super().__init__(num_channels=num_channels, super().__init__(num_channels=num_channels,
sampling_rate=sampling_rate,lr=lr, sampling_rate=sampling_rate,lr=lr,
dataset=dataset,duration=duration,loss=loss, metric=metric) dataset=dataset,duration=duration,loss=loss, metric=metric)

8
enhancer/models/waveunet.py
View File

@ -1,5 +1,4 @@
from tkinter import wantobjects import logging
import wave
import torch import torch
import torch.nn as nn import torch.nn as nn
import torch.nn.functional as F import torch.nn.functional as F
@ -71,7 +70,10 @@ class WaveUnet(Model):
metric:Union[str,List] = "mse" metric:Union[str,List] = "mse"
): ):
duration = dataset.duration if isinstance(dataset,EnhancerDataset) else None duration = dataset.duration if isinstance(dataset,EnhancerDataset) else None
sampling_rate = sampling_rate if dataset is None else dataset.sampling_rate if dataset is not None:
if sampling_rate!=dataset.sampling_rate:
logging.warn(f"model sampling rate {sampling_rate} should match dataset sampling rate {dataset.sampling_rate}")
sampling_rate = dataset.sampling_rate
super().__init__(num_channels=num_channels, super().__init__(num_channels=num_channels,
sampling_rate=sampling_rate,lr=lr, sampling_rate=sampling_rate,lr=lr,
dataset=dataset,duration=duration,loss=loss, metric=metric dataset=dataset,duration=duration,loss=loss, metric=metric