fix setter
This commit is contained in:
parent
8c9511967c
commit
124ee83180
|
|
@ -74,9 +74,9 @@ class Model(pl.LightningModule):
|
||||||
def loss(self, loss):
|
def loss(self, loss):
|
||||||
|
|
||||||
if isinstance(loss, str):
|
if isinstance(loss, str):
|
||||||
losses = [loss]
|
loss = [loss]
|
||||||
|
|
||||||
self._loss = Avergeloss(losses)
|
self._loss = Avergeloss(loss)
|
||||||
|
|
||||||
@property
|
@property
|
||||||
def metric(self):
|
def metric(self):
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue