Skip to content

S
seminar-breakout
Commit e891b9e7 authored by Yuxin Wu's avatar Yuxin Wu
Browse files
Options

Add AccumGradOptimizer (#391)

parent 653b4b97
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 1 deletion
tensorpack/tfutils/optimizer.py
View file @ e891b9e7
...@@ -8,7 +8,8 @@ from contextlib import contextmanager ...@@ -8,7 +8,8 @@ from contextlib import contextmanager
from .gradproc import FilterNoneGrad from .gradproc import FilterNoneGrad
__all__ = ['apply_grad_processors', 'ProxyOptimizer', __all__ = ['apply_grad_processors', 'ProxyOptimizer',
'PostProcessOptimizer', 'VariableAssignmentOptimizer'] 'PostProcessOptimizer', 'VariableAssignmentOptimizer',
'AccumGradOptimizer']
class ProxyOptimizer(tf.train.Optimizer): class ProxyOptimizer(tf.train.Optimizer):
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment