onnxruntime/orttraining/orttraining/python/training/optim
Xavier Dupré b508c7236f
Replace call to deprecated torch.norm (#16758)
### Description
torch.norm is deprecated as mentioned in issue #16751. This PR replaces
the call to torch.norm by the options suggested by torch documentation.
2023-07-20 19:52:19 -07:00
..
__init__.py Adopt linrtunner as the linting tool - take 2 (#15085) 2023-03-24 15:29:03 -07:00
_apex_amp_modifier.py Enable pylint and numpy rules (#15218) 2023-03-27 20:37:53 -07:00
_ds_modifier.py support latest deepspeed version for optim (#15682) 2023-04-25 20:12:23 -07:00
_megatron_modifier.py
_modifier.py Replace call to deprecated torch.norm (#16758) 2023-07-20 19:52:19 -07:00
_modifier_registry.py Adopt linrtunner as the linting tool - take 2 (#15085) 2023-03-24 15:29:03 -07:00
_multi_tensor_apply.py Adopt linrtunner as the linting tool - take 2 (#15085) 2023-03-24 15:29:03 -07:00
config.py Bump ruff in CI (#15533) 2023-04-17 10:11:44 -07:00
fp16_optimizer.py Adopt linrtunner as the linting tool - take 2 (#15085) 2023-03-24 15:29:03 -07:00
fused_adam.py Adding this set_to_none flag to zero_grad to have signature parity with pytorch Adam (#16375) 2023-06-19 17:27:41 -07:00
lr_scheduler.py Adopt linrtunner as the linting tool - take 2 (#15085) 2023-03-24 15:29:03 -07:00