Skip to content

Commit

Permalink
fix: it's starting to get frustrating
Browse files Browse the repository at this point in the history
  • Loading branch information
BrunoLiegiBastonLiegi committed Nov 22, 2024
1 parent 170c63c commit 04632be
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion tests/test_models_interfaces.py
Original file line number Diff line number Diff line change
Expand Up @@ -200,7 +200,7 @@ def backprop_test(frontend, model, data, target):
grad = train_model(frontend, model, data, target)
_, loss_trained = eval_model(frontend, model, data, target)
assert grad < 1e-2
assert loss_untrained >= loss_trained
assert round(float(loss_untrained), 6) >= round(float(loss_trained), 6)
# in some (unpredictable) cases the gradient and loss
# start so small that the model doesn't do anything
# fixing the seed doesn't fix this on all the platforms
Expand Down

0 comments on commit 04632be

Please sign in to comment.