fix setter

This commit is contained in:
shahules786 2022-10-06 11:30:00 +05:30
parent 8c9511967c
commit 124ee83180
1 changed files with 2 additions and 2 deletions

View File

@ -74,9 +74,9 @@ class Model(pl.LightningModule):
def loss(self, loss):
if isinstance(loss, str):
losses = [loss]
loss = [loss]
self._loss = Avergeloss(losses)
self._loss = Avergeloss(loss)
@property
def metric(self):