3D-UNet (KiTS19) Pytorch model not processed by jit.trace

mlcommons / inference

Reference implementations of MLPerf™ inference benchmarks

https://mlcommons.org/en/groups/inference

Apache License 2.0

1.22k stars 532 forks source link

3D-UNet (KiTS19) Pytorch model not processed by jit.trace #1031

Closed rnaidu02 closed 2 years ago

rnaidu02 commented 3 years ago

Current 3D-UNet model (KiTS19) is already processed by jit.trace. Since it is already jitted, it is not possible for performing jit.trace for platform specific optimizations. Intel team is requesting reference implementation owner to share the model not processed by jit.trace (original model weights).
@delock

DilipSequeira commented 3 years ago

@rnaidu02 please review https://github.com/mlcommons/inference/pull/1033

xuwh-intel commented 2 years ago

Thanks, this PR works for me.

rnaidu02 commented 2 years ago

Thanks @DilipSequeira