未验证 提交 bcb03531 编写于 作者: S Stas Bekman 提交者: GitHub

Update engine.py (#2826)

Co-authored-by: NOlatunji Ruwase <olruwase@microsoft.com>
上级 6ba0024d
......@@ -112,7 +112,6 @@ try:
except ImportError:
# Fail silently so we don't spam logs unnecessarily if user isn't using amp
APEX_INSTALLED = False
pass
def split_half_float_double_sparse(tensors):
......@@ -1702,7 +1701,6 @@ class DeepSpeedEngine(Module):
# we are in a forward pass.
for module in self.module.modules():
module._parameters._in_forward = True
pass
self._start_timers(self.engine_timers.forward_timers)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册