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

4
enhancer/models/model.py
View File

@ -3,7 +3,7 @@ from torch.optim import Adam
import pytorch_lightning as pl import pytorch_lightning as pl
from enhancer.data.dataset import Dataset from enhancer.data.dataset import Dataset
from enhancer.utils.loss import LOSS_MAP, Avergeloss from enhancer.utils.loss import Avergeloss
class Model(pl.LightningModule): class Model(pl.LightningModule):
@ -51,7 +51,7 @@ class Model(pl.LightningModule):
return self.dataset.val_dataloader() return self.dataset.val_dataloader()
def configure_optimizers(self): 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): def training_step(self,batch, batch_idx:int):