From 3e98e2ab20ba9c68beab7ad02a484d8a5721d983 Mon Sep 17 00:00:00 2001 From: Mo Zhou Date: Mon, 5 Dec 2022 16:00:57 -0500 Subject: [PATCH] trivial: silent --- robrank/defenses/amd.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/robrank/defenses/amd.py b/robrank/defenses/amd.py index 5f8db00..8a1b7fd 100644 --- 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