Code

Add ONNX exporter#1826

Code

Open

MathijsdeBoer wants to merge 14 commits intoMIC-DKFZ:masterMIC-DKFZ/nnUNet:masterfrom MathijsdeBoer:masterMathijsdeBoer/nnUNet:masterCopy head branch name to clipboard

+384-35

Commits

Commits on Nov 30, 2023

• Initial implementation of onnx export

  

  Mathijs de Boer

  committed
  c57ddfa

Commits on Dec 1, 2023

• Finalizing a few things

  

  Mathijs de Boer

  committed
  88ae8e2

• Add ONNX exporter, no guarantees are provided for

  Show description for 7f4b854
  

  Mathijs de Boer

  committed
  7f4b854

• Merge remote-tracking branch 'origin/master'

  

  Mathijs de Boer

  committed
  e374385

• Cleanup entry_points.py

  

  Mathijs de Boer

  committed
  438e633

• Switch from dynamo export to regular export

  Show description for 979a716
  

  Mathijs de Boer

  committed
  979a716

• Hit a recursion limit bug, increase

  

  Mathijs de Boer

  committed
  1d6983c

• Currently throws a segmentation fault, disabling compile for the script duration

  

  Mathijs de Boer

  committed
  bec25b5

• Torch was unable to set instance_norm trainmode to False, this might be causing the asser_allclose to fail. Make it a warning for now.

  

  Mathijs de Boer

  committed
  f4dfc3c

Commits on Dec 4, 2023

• Add some more information to the config.json

  

  Mathijs de Boer

  committed
  7487021

• Forgot to write the new config

  

  Mathijs de Boer

  committed
  fef8334

• Add method to fix batch size, for potential speedups. Default still dynamic sized.

  Show description for a190898
  

  Mathijs de Boer

  committed
  a190898

Commits on Dec 5, 2023

• Add foreground statistics for when normalization is CT

  

  Mathijs de Boer

  committed
  4507cb4

Commits on Dec 14, 2023

• Update warning text

  

  Mathijs de Boer

  committed
  4fa40e6