Vincent Wang 67150baa8d [ORTModule] ATen Support for aten::upsample_nearest (#13364) ... ATen support for aten::upsample_nearest, which is required for Huggingface's diffusers model training using ORTModule.		2022-10-20 08:30:04 +08:00
..
amp	Format all python files under onnxruntime with black and isort (#11324)	2022-04-26 09:35:16 -07:00
api	Add Utils for federated learning scenarios (#13014)	2022-10-17 12:39:43 -07:00
experimental	Make ORT callable from various Pytorch compilers (LazyTensor, TorchDynamo, etc) (#10460)	2022-08-22 09:40:40 -07:00
onnxblock	Use InplaceClipGradNorm for offline processing for on-device training (#12603)	2022-09-02 07:47:17 -07:00
optim	[ORTModule] Update Supported DeepSpeed Version for FP16_Optimizer (#13305)	2022-10-13 13:03:01 +08:00
ortmodule	[ORTModule] ATen Support for aten::upsample_nearest (#13364)	2022-10-20 08:30:04 +08:00
utils/data	Format all python files under onnxruntime with black and isort (#11324)	2022-04-26 09:35:16 -07:00
__init__.py	[python] [orttraining] Add utility to export a graph to compute gradients (#8125)	2022-02-18 14:00:49 -08:00
_checkpoint_storage.py	Format all python files under onnxruntime with black and isort (#11324)	2022-04-26 09:35:16 -07:00
_utils.py	replace 'master' branch ref to 'main' for onnx repo (#12678)	2022-08-30 13:41:42 -07:00
checkpoint.py	Format all python files under onnxruntime with black and isort (#11324)	2022-04-26 09:35:16 -07:00
model_desc_validation.py	Format all python files under onnxruntime with black and isort (#11324)	2022-04-26 09:35:16 -07:00
orttrainer.py	Deprecate ORTTrainer (#13022)	2022-09-23 18:10:09 -07:00
orttrainer_options.py	Update ORTModule Default Opset Version to 15 (#12419)	2022-08-05 16:55:04 +08:00
postprocess.py	Format all python files under onnxruntime with black and isort (#11324)	2022-04-26 09:35:16 -07:00