Merge branch 'master' of https://github.com/SSARCandy/DeepCORAL

README.md

@@ -1,6 +1,9 @@
 # Deep CORAL
 
-An PyTorch implementation of 'Deep CORAL: Correlation Alignment for Deep Domain Adaptation. B Sun, K Saenko, ECCV 2016'
+A PyTorch implementation of 'Deep CORAL: Correlation Alignment for Deep Domain Adaptation. B Sun, K Saenko, ECCV 2016'
+
+> Deep CORAL can learn a nonlinear transformation that aligns correlations of
+layer activations in deep neural networks (Deep CORAL).
 
 [中文介紹](https://ssarcandy.tw/2017/10/31/deep-coral/)
 
@@ -20,7 +23,19 @@ My implementation result (Task Amazon -> Webcam):
 
 ## Unit Test
 
+The unit test is for sanity test only (to check my math is not going wrong),
+please run it under CPU version of PyTorch.
+
 ```bash
 $ cd ..
 $ python -m DeepCORAL.tests.test -v
-```
+test_CORAL_backward (__main__.TestCORAL) ... ok
+test_CORAL_forward (__main__.TestCORAL) ... ok
+test_feature_covariance_mat (__main__.TestCORAL) ... ok
+test_forbenius_norm (__main__.TestCORAL) ... ok
+
+----------------------------------------------------------------------
+Ran 4 tests in 0.033s
+
+OK
+```