lightning/pytorch_lightning
Lezwon Castelino b7afac351b
Add onnx export (#2596)
* export model to onnx

* prepare data before exporting

* support for dataloaders and tensors

* added tests

* use example_input_array
add to changelog

* updated docstring

* added onnx inference tests

* temp commit

* removed schema valid test

* add onnxruntime to environment.yml

* moved onnxruntime to environment.yml pip

* add example in doc

* add lines between code block

* added PR to changelog

* is file check

Co-authored-by: Jirka Borovec <Borda@users.noreply.github.com>

* remove *

Co-authored-by: Jirka Borovec <Borda@users.noreply.github.com>

* infer example outputs

* added doctest for onnx

* fix windows tests

* moved eval within condition block

* self.forward to self

* added docs

* fixed docs error

* added to toctree

* Update CHANGELOG.md

Co-authored-by: Jirka Borovec <Borda@users.noreply.github.com>
2020-07-31 12:27:57 +02:00
..
accelerator_backends fixing TPU tests (#2632) 2020-07-27 19:07:09 -04:00
callbacks Fix docs typo (#2747) 2020-07-29 07:11:49 -04:00
core Add onnx export (#2596) 2020-07-31 12:27:57 +02:00
loggers Add missing methods to logger collection (#2723) 2020-07-29 23:53:02 +02:00
metrics metrics: add SSIM (#2671) 2020-07-23 12:13:52 -04:00
overrides Structured results (train loop only. val loop separate PR) (PR 2/5) (#2615) 2020-07-20 19:00:20 -04:00
profiler replace Hparams by init args (#1896) 2020-05-24 18:59:08 -04:00
trainer pytorch 1.6 (#2745) 2020-07-31 11:18:32 +02:00
utilities Fix a deprecation warning (#2746) 2020-07-29 07:54:14 -04:00
__init__.py remove deprecated in v0.9 (#2760) 2020-07-30 23:19:28 +02:00