|
|
@@ -74,5 +74,3 @@ def test_ftrl(): |
|
|
label = Tensor(np.array([0]).astype(np.int32)) |
|
|
label = Tensor(np.array([0]).astype(np.int32)) |
|
|
loss = train_network(data, label) |
|
|
loss = train_network(data, label) |
|
|
losses2.append(loss.asnumpy()) |
|
|
losses2.append(loss.asnumpy()) |
|
|
assert losses2[0] > losses2[1] |
|
|
|
|
|
assert losses2[1] > losses2[2] |
|
|
|