trivial: silent

cdluminate · Dec 5, 2022 · 3e98e2a · 3e98e2a
1 parent 30cd8c5
commit 3e98e2a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/robrank/defenses/amd.py b/robrank/defenses/amd.py
@@ -785,7 +785,7 @@ def hm_training_step(model: th.nn.Module, batch, batch_idx, *,
         loss_bt = barlow.barlow_twins(pnemb[:len(pnemb)//3], labels[anc],
                   pnemb[len(pnemb)//3:2*len(pnemb)//3], labels[pos])
         model.log('Train/loss_bt', loss_bt.item())
-        print('loss_bt', loss_bt.item())
+        #print('loss_bt', loss_bt.item())
         loss = loss + 1e-5 * loss_bt
     # return
     return loss