fix setter
This commit is contained in:
parent
8c9511967c
commit
124ee83180
|
|
@ -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):
|
||||
|
|
|
|||
Loading…
Reference in New Issue