Searched refs:GradientAccumulation (Results 1 – 4 of 4) sorted by relevance
45 class GradientAccumulation(Cell): class54 super(GradientAccumulation, self).__init__()
27 from .grad_accumulation import GradientAccumulation
28 from .grad_accumulation import GradientAccumulation144 self.grad_accumulation = GradientAccumulation(
117 class GradientAccumulation: class217 model = GradientAccumulation(network, net_loss, net_opt)