Update emlloss.py
parent
adfd5c4c10
commit
e9a1d83035
|
@ -87,14 +87,3 @@ class EmlLoss(paddle.nn.Layer):
|
||||||
loss = paddle.mean(loss)
|
loss = paddle.mean(loss)
|
||||||
return {"emlloss": loss}
|
return {"emlloss": loss}
|
||||||
|
|
||||||
if __name__=="__main__":
|
|
||||||
|
|
||||||
metric = EmlLoss()
|
|
||||||
|
|
||||||
np.random.seed(1)
|
|
||||||
features = np.random.randn(40, 32)
|
|
||||||
features = paddle.to_tensor(features, dtype="float32")
|
|
||||||
print(features)
|
|
||||||
|
|
||||||
loss = metric(features)
|
|
||||||
print(loss)
|
|
Loading…
Reference in New Issue