Code

Add onnx export#2596

Code

Merged

Borda merged 25 commits intoLightning-AI:masterLightning-AI/pytorch-lightning:masterfrom lezwon:feature/2271_onnx_exportlezwon/pytorch-lightning:feature/2271_onnx_exportCopy head branch name to clipboardJul 31, 2020

Commits

Commits on Jul 29, 2020

• export model to onnx

  

  lezwon

  committed
  88a9eda

  Copy full SHA for 88a9eda

• prepare data before exporting

  

  lezwon

  committed
  29c330e

  Copy full SHA for 29c330e

• support for dataloaders and tensors

  

  lezwon

  committed
  c154e4a

  Copy full SHA for c154e4a

• added tests

  

  lezwon

  committed
  08f75e7

  Copy full SHA for 08f75e7

• use example_input_array

  Show description for 61cf39f
  

  lezwon

  committed
  61cf39f

  Copy full SHA for 61cf39f

• updated docstring

  

  lezwon

  committed
  feb7b56

  Copy full SHA for feb7b56

• added onnx inference tests

  

  lezwon

  committed
  81cc4a0

  Copy full SHA for 81cc4a0

• temp commit

  

  lezwon

  committed
  cb92d28

  Copy full SHA for cb92d28

• removed schema valid test

  

  lezwon

  committed
  4f2aadc

  Copy full SHA for 4f2aadc

• add onnxruntime to environment.yml

  

  lezwon

  committed
  0289f2b

  Copy full SHA for 0289f2b

• moved onnxruntime to environment.yml pip

  

  lezwon

  committed
  9a8b129

  Copy full SHA for 9a8b129

• add example in doc

  

  lezwon

  committed
  5265578

  Copy full SHA for 5265578

• add lines between code block

  

  lezwon

  committed
  de3b028

  Copy full SHA for de3b028

• added PR to changelog

  

  lezwon

  committed
  467eba1

  Copy full SHA for 467eba1

• is file check

  Show description for d5e3eb6
  

  2 peopleauthored and

  lezwon

  committed
  d5e3eb6

  Copy full SHA for d5e3eb6

• remove *

  Show description for a6065fb
  

  2 peopleauthored and

  lezwon

  committed
  a6065fb

  Copy full SHA for a6065fb

• infer example outputs

  

  lezwon

  committed
  0fb4b28

  Copy full SHA for 0fb4b28

• added doctest for onnx

  

  lezwon

  committed
  49982a3

  Copy full SHA for 49982a3

• fix windows tests

  

  lezwon

  committed
  434c8d8

  Copy full SHA for 434c8d8

• moved eval within condition block

  

  lezwon

  committed
  b7ead43

  Copy full SHA for b7ead43

• self.forward to self

  

  lezwon

  committed
  633a47c

  Copy full SHA for 633a47c

• added docs

  

  lezwon

  committed
  add43b7

  Copy full SHA for add43b7

• fixed docs error

  

  lezwon

  committed
  d71f744

  Copy full SHA for d71f744

• added to toctree

  

  lezwon

  committed
  cd5df3f

  Copy full SHA for cd5df3f

Commits on Jul 31, 2020

• Update CHANGELOG.md

  Borda

  authored
  dcc0ad9

  Copy full SHA for dcc0ad9