fix loss

close #14
SSARCandy · Mar 13, 2018 · 6f91c19 · 6f91c19
1 parent c6f55b6
commit 6f91c19
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/models.py b/models.py
@@ -23,7 +23,7 @@ def CORAL(source, target):
 
     # frobenius norm between source and target
     loss = torch.mean(torch.mul((xc - xct), (xc - xct)))
-    loss = loss/(4*d*4)
+    loss = loss/(4*d*d)
 
     return loss