fix model sr to dataset sr

This commit is contained in:
shahules786 2022-09-29 12:27:04 +05:30
parent 18759a3f84
commit e22cecaf20
1 changed files with 4 additions and 0 deletions

View File

@ -1,5 +1,6 @@
from importlib import import_module
from huggingface_hub import cached_download, hf_hub_url
import logging
import numpy as np
import os
from typing import Optional, Union, List, Text, Dict, Any
@ -37,6 +38,9 @@ class Model(pl.LightningModule):
super().__init__()
assert num_channels ==1 , "Enhancer only support for mono channel models"
self.dataset = dataset
if self.dataset is not None:
sampling_rate = self.dataset.sampling_rate
logging.warn("Setting model sampling rate same as dataset sampling rate")
self.save_hyperparameters("num_channels","sampling_rate","lr","loss","metric","duration")
if self.logger:
self.logger.experiment.log_dict(dict(self.hparams),"hyperparameters.json")