Skip to content

Commit eddce17

Browse files
committed
Docstring for get_stat_updates.
1 parent 955eaa2 commit eddce17

File tree

1 file changed

+7
-0
lines changed

1 file changed

+7
-0
lines changed

DeepFried2/Module.py

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -99,6 +99,13 @@ def accumulate_gradients(self, data_in, data_tgt, loss):
9999
return self._fn_accum_grads[self.training_mode](*args)
100100

101101
def get_stat_updates(self):
102+
"""
103+
Return extra `update` statements, currently only Batch-Normalization.
104+
As soon as something else has a similar need, we might need to unify
105+
them, and possibly change/generalize the training/evaluate modes.
106+
107+
Guaranteed to be called after `symb_forward`.
108+
"""
102109
return []
103110

104111
def accumulate_statistics(self, data_in):

0 commit comments

Comments
 (0)