X-Git-Url: https://fleuret.org/cgi-bin/gitweb/gitweb.cgi?a=blobdiff_plain;f=tasks.py;h=24c13fe79eddb4b8104bfe869c079bb4cbce80b9;hb=5332c56acd44d7049f3fbb33a8643482e0c71f4d;hp=2c2f9140ec337ae5cac4c490a40d97a92fa54fe1;hpb=0e1e208852b83f6a3d59e5caabd2f0f1f4bde94e;p=picoclvr.git diff --git a/tasks.py b/tasks.py index 2c2f914..24c13fe 100755 --- a/tasks.py +++ b/tasks.py @@ -1539,8 +1539,8 @@ class Grid(Task): nb_total = ar_mask.sum().item() nb_correct = ((correct == result).long() * ar_mask).sum().item() - logger(f"test_performance {nb_total=} {nb_correct=}") - logger(f"main_test_accuracy {nb_correct / nb_total}") + logger(f"test_performance {n_epoch} {nb_total=} {nb_correct=}") + logger(f"main_test_accuracy {n_epoch} {nb_correct / nb_total}") ######################################################################