doc for row-wise adagrad optimizer

jianan-gu · Dec 18, 2020 · 3593be9 · 3593be9
1 parent 55becd8
commit 3593be9
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 1 deletion.
diff --git a/README.md b/README.md
@@ -364,7 +364,7 @@ Version
 -------
 0.1 : Initial release of the DLRM code
 
-1.0 : DLRM with distributed training
+1.0 : DLRM with distributed training, cpu support for row-wise adagrad optimizer
 
 Requirements
 ------------

diff --git a/optim/rwsadagrad.py b/optim/rwsadagrad.py
@@ -1,3 +1,9 @@
+# Copyright (c) Facebook, Inc. and its affiliates.
+#
+# This source code is licensed under the MIT license found in the
+# LICENSE file in the root directory of this source tree.
+
+
 import torch
 from torch.optim import Optimizer