Skip to content

Commit 8696614

Browse files
committed
Update version to 0.1.5
1 parent 338873e commit 8696614

File tree

2 files changed

+13
-1
lines changed

2 files changed

+13
-1
lines changed

README.md

+12
Original file line numberDiff line numberDiff line change
@@ -428,6 +428,18 @@ init = ZeroLastGamma(block_name='Bottleneck', bn_name='bn3')
428428
model.apply(init)
429429
```
430430

431+
#### 15. SGC_GC
432+
[Gradient Centralization: A New Optimization Technique for Deep Neural Networks](https://arxiv.org/pdf/2004.01461.pdf)
433+
434+
```python
435+
from torchtoolbox.optimizer import SGD_GC
436+
437+
optimizer = SGD_GC(model.parameters())
438+
optimizer.zero_grad()
439+
loss.backward()
440+
optimizer.step()
441+
442+
```
431443

432444
## Contribution
433445

torchtoolbox/__init__.py

100644100755
+1-1
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
11
# -*- coding: utf-8 -*-
22
33

4-
VERSION = '0.1.4.1'
4+
VERSION = '0.1.5'

0 commit comments

Comments
 (0)