Skip to content

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

bugfix in periodic trigger

parent 1cd4a380
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
tensorpack/callbacks/trigger.py
View file @ c1a280a2
...@@ -27,8 +27,8 @@ class PeriodicTrigger(ProxyCallback): ...@@ -27,8 +27,8 @@ class PeriodicTrigger(ProxyCallback):
super(PeriodicTrigger, self).__init__(triggerable) super(PeriodicTrigger, self).__init__(triggerable)
assert (every_k_epochs is not None) or (every_k_steps is not None), \ assert (every_k_epochs is not None) or (every_k_steps is not None), \
"every_k_steps and every_k_epochs cannot be both None!" "every_k_steps and every_k_epochs cannot be both None!"
self._step_k = int(every_k_steps) self._step_k = every_k_steps
self._epoch_k = int(every_k_epochs) self._epoch_k = every_k_epochs
def _trigger_step(self): def _trigger_step(self):
if self._step_k is None: if self._step_k is None:
......
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