add CL

2019-04-03 18:36:47 +08:00 · 2019-04-03 18:36:47 +08:00 · 16b2d15d5b
parent 98fe18e5f9
commit 16b2d15d5b
2 changed files with 1 additions and 0 deletions
--- a/.DS_Store
+++ b/.DS_Store
--- a/utils/train_util_DCL.py
+++ b/utils/train_util_DCL.py
@ -59,6 +59,7 @@ def train(cfg,
            if isinstance(outputs, list):
                loss = criterion(outputs[0], labels)
                loss += criterion(outputs[1], labels_swap)
+                loss += add_loss(outputs[2], swap_law)
            loss.backward()
            optimizer.step()