onnxruntime/orttraining
Ryota Tomioka 9835b46a1d
Add an option to save the training graph after optimization (#6410)
* expose optimized_model_filepath in SessionOptions as `debug.graph_save_paths.model_with_training_graph_after_optimization_path` in `ORTTrainerOptions`
2021-01-27 07:39:46 +00:00
..
orttraining Add an option to save the training graph after optimization (#6410) 2021-01-27 07:39:46 +00:00
pytorch_frontend_examples Enable running the mnist_training sample without cuda (#6085) 2020-12-15 17:06:54 -08:00
tools Dockerfile to build onnxruntime with ROCm 4.0 2020-12-22 10:21:12 -08:00