add CL

JDAI-CV · Apr 3, 2019 · 16b2d15 · 16b2d15
1 parent 98fe18e
commit 16b2d15
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 0 deletions.
diff --git a/.DS_Store b/.DS_Store
diff --git 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()