Update version to 0.1.5

PistonY · Aug 3, 2020 · 8696614 · 8696614
1 parent 338873e
commit 8696614
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 1 deletion.
diff --git a/README.md b/README.md
@@ -428,6 +428,18 @@ init = ZeroLastGamma(block_name='Bottleneck', bn_name='bn3')
 model.apply(init)
 ```
 
+#### 15. SGC_GC
+[Gradient Centralization: A New Optimization Technique for Deep Neural Networks](https://arxiv.org/pdf/2004.01461.pdf)
+
+```python
+from torchtoolbox.optimizer import SGD_GC
+
+optimizer = SGD_GC(model.parameters())
+optimizer.zero_grad()
+loss.backward()
+optimizer.step()
+
+```
 
 ## Contribution
 

diff --git a/torchtoolbox/__init__.py b/torchtoolbox/__init__.py
@@ -1,4 +1,4 @@
 # -*- coding: utf-8 -*-
 # Author: pistonyang@gmail.com
 
-VERSION = '0.1.4.1'
+VERSION = '0.1.5'