saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-06-27 03:11:28 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Update Supporting DS Version to 0.7.1 for ORTModule (#12696)

Browse source 
update ds version support for fp16_optimizer
This commit is contained in:
Vincent Wang 2022-08-24 14:56:12 +08:00 committed by GitHub
parent de3d772995
commit 53ecb9e635
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
orttraining/orttraining/python/training/optim/_ds_modifier.py
View file

@ -31,7 +31,7 @@ class DeepSpeedZeROModifier(FP16OptimizerModifier):
import deepspeed
ds_version = LooseVersion(deepspeed.__version__)
if ds_version > LooseVersion("0.7.0") or ds_version < LooseVersion("0.4.0"):
if ds_version > LooseVersion("0.7.1") or ds_version < LooseVersion("0.4.0"):
warnings.warn("Skip modifying optimizer because of unsupported DeepSpeed version.", UserWarning)
return False