BaguaSys / bagua

Bagua Speeds up PyTorch

Home Page:https://tutorials.baguasys.com/

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

`fused_step` may miss `init_post_optimizer_step_hook`

todo opened this issue · comments

commented

# TODO: `fused_step` may miss `init_post_optimizer_step_hook`
def new_step_factory(optimizer):
def new_step(self, *args, **kwargs):
result = self._bagua_original_step(*args, **kwargs)


This issue was generated by todo based on a TODO comment in 5add7f8 when #207 was merged. cc @BaguaSys.