mirror of https://github.com/InternLM/InternLM
parent
f653e5af01
commit
42e5f6f8a9
|
@ -709,7 +709,7 @@ class HybridZeroOptimizer(BaseOptimizer):
|
|||
for handle in handles:
|
||||
handle.wait()
|
||||
|
||||
torch.cuda().synchronize()
|
||||
torch.cuda.synchronize()
|
||||
|
||||
##################
|
||||
# FP16 Utilities #
|
||||
|
|
Loading…
Reference in New Issue