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

remove unused imports

This commit is contained in:
shahules786 2022-09-12 10:55:24 +05:30
parent 9ed1b9d3f7
commit 4536a82e90
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
enhancer/models/model.py
View File

@ -3,7 +3,7 @@ from torch.optim import Adam
import pytorch_lightning as pl
from enhancer.data.dataset import Dataset
from enhancer.utils.loss import LOSS_MAP, Avergeloss
from enhancer.utils.loss import Avergeloss
class Model(pl.LightningModule):
@ -20,7 +20,7 @@ class Model(pl.LightningModule):
assert num_channels ==1 , "Enhancer only support for mono channel models"
self.save_hyperparameters("num_channels","sampling_rate","lr","loss")
self.dataset = dataset
@property
def dataset(self):
@ -51,7 +51,7 @@ class Model(pl.LightningModule):
return self.dataset.val_dataloader()
def configure_optimizers(self):
return Adam(self.parameters, lr = self.hparams.lr)
return Adam(self.parameters(), lr = self.hparams.lr)
def training_step(self,batch, batch_idx:int):