Update emlloss.py
parent
adfd5c4c10
commit
e9a1d83035
|
@ -87,14 +87,3 @@ class EmlLoss(paddle.nn.Layer):
|
|||
loss = paddle.mean(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