[ONNX Import] Asymmetric padding is not supported

[justinchuby]

I get

  ERROR burn_import::logger: PANIC => panicked at /Users/justinc/.cargo/registry/src/index.crates.io-1949cf8c6b5b557f/onnx-ir-0.19.1/src/node/padding.rs:124:9:
  Asymmetric padding is not supported

When importing https://huggingface.co/justinchuby/BirdNET-onnx/blob/main/birdnet.onnx. The error is also unclear in which onnx node caused the issue.

[antimora]

We added more padding options in #4105

If you could try from the main, it should work.

[justinchuby]

I am still getting the same error from main:

Caused by:
  process didn't exit successfully: `/Users/justinc/Documents/GitHub/my_burn_app/target/debug/build/my_burn_app-ba40587427de64ca/build-script-build` (exit status: 101)
  --- stdout
   INFO burn_import::onnx::model_gen: Starting to convert ONNX to Burn
  DEBUG burn_import::onnx::model_gen: Output directory: "/Users/justinc/Documents/GitHub/my_burn_app/target/debug/build/my_burn_app-20904dfebb30007b/out/model/"
   INFO burn_import::onnx::model_gen: Converting "/Users/justinc/Documents/GitHub/BirdNET-onnx/birdnet.onnx"
  DEBUG burn_import::onnx::model_gen: Input file name: "birdnet"
  DEBUG burn_import::onnx::model_gen: Output file: "/Users/justinc/Documents/GitHub/my_burn_app/target/debug/build/my_burn_app-20904dfebb30007b/out/model/birdnet"
   INFO burn_import::onnx::model_gen: Generating model from "/Users/justinc/Documents/GitHub/BirdNET-onnx/birdnet.onnx"
  DEBUG burn_import::onnx::model_gen: Development mode: false
  DEBUG burn_import::onnx::model_gen: Output file: "/Users/justinc/Documents/GitHub/my_burn_app/target/debug/build/my_burn_app-20904dfebb30007b/out/model/birdnet"
   INFO onnx_ir::pipeline: Parsing ONNX file: /Users/justinc/Documents/GitHub/BirdNET-onnx/birdnet.onnx
  DEBUG onnx_ir::pipeline: Memory-mapped ONNX file (66932288 bytes)
  DEBUG onnx_ir::pipeline:  PHASE 1: Initialization 
  DEBUG onnx_ir::pipeline:  PHASE 2: Node Conversion (Proto -> RawNode) 
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant1_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant1_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant1_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant1_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant2_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant2_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant3_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant3_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant4_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant4_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant5_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant5_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant6_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant6_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant7_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant7_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant8_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant8_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant9_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant9_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant10_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant10_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant11_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant11_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant12_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant12_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant13_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Converting MatMul → Linear for node matmul1
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant13_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant14_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Converting MatMul → Linear for node matmul2
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant14_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant15_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Converting MatMul → Linear for node matmul3
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant15_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant16_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Converting MatMul → Linear for node matmul4
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant16_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant17_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant17_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant18_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant18_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant19_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant20_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant19_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant19_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant19_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant20_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant19_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant19_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant19_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant20_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant19_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant19_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant19_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant20_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant19_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant19_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant19_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant19_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant19_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant19_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant21_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant21_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant22_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant22_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant23_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant24_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant23_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant24_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant25_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant26_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant25_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant26_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant27_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant28_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant27_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant28_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant29_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant30_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant29_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant30_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant31_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant31_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant32_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant33_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant32_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant33_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant34_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant35_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant34_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant35_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant36_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant36_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant37_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant38_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant37_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant38_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant39_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant40_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant39_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant40_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant41_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant41_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant42_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant43_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant42_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant43_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant44_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant45_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant44_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant45_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant46_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant46_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant47_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant47_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant48_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant48_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant49_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant49_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant50_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant50_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant51_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant51_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant52_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant53_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant52_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant53_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant54_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant55_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant54_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant55_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant46_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant46_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant47_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant47_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant56_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant56_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant49_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant49_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant57_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant57_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant58_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant58_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant59_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant60_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant59_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant60_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant61_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant62_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant61_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant62_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant46_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant46_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant47_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant47_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant63_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant63_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant49_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant49_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant64_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant64_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant65_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant65_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant66_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant67_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant66_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant67_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant68_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant69_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant68_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant69_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant46_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant46_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant47_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant47_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant70_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant70_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant49_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant49_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant71_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant71_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant72_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant72_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant73_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant74_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant73_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant74_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant75_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant76_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant75_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant76_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant46_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant46_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant77_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant77_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant78_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant78_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant79_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant79_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant80_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant80_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant81_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant81_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant82_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant83_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant82_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant83_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant84_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant85_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant84_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant85_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant46_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant46_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant77_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant77_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant86_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant86_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant79_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant79_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant87_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant87_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant88_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant88_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant89_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant90_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant89_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant90_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant91_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant92_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant91_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant92_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant46_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant46_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant77_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant77_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant93_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant93_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant79_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant79_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant94_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant94_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant95_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant95_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant96_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant97_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant96_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant97_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant98_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant99_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant98_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant99_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant46_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant46_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant77_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant77_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant100_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant100_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant79_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant79_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant101_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant101_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant102_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant102_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant103_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant104_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant103_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant104_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant105_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant106_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant105_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant106_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant46_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant46_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant77_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant77_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant107_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant107_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant79_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant79_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant108_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant108_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant109_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant109_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant110_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant111_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant110_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant111_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant112_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant113_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant112_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant113_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant46_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant46_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant114_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant114_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant115_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant115_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant116_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant116_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant117_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant117_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant118_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant118_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant119_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant120_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant119_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant120_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant121_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant122_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant121_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant122_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant46_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant46_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant114_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant114_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant123_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant123_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant116_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant116_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant124_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant124_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant125_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant125_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant126_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant127_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant126_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant127_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant128_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant129_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant128_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant129_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant46_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant46_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant114_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant114_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant130_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant130_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant116_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant116_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant131_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant131_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant132_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant132_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant133_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant134_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant133_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant134_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant135_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant136_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant135_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant136_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant46_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant46_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant114_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant114_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant137_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant137_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant116_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant116_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant138_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant138_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant139_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant139_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant140_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant140_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant141_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant141_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant142_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant143_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant142_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant143_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant46_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant46_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant144_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Converting MatMul → Linear for node matmul5
   WARN onnx_ir::graph_state: Input Identity_Y_0 not found, should only happen when peeking
  DEBUG onnx_ir::phases::node_conversion: Fused Add bias into Linear node matmul5
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant144_out1' (Constant)
  DEBUG onnx_ir::phases::node_conversion: Preserving outer-scope value_store for 'constant145_out1' (Constant)
  DEBUG onnx_ir::pipeline:  PHASE 3: Type Inference 
  DEBUG onnx_ir::phases::type_inference: Type inference converged after 2 iterations
  DEBUG onnx_ir::pipeline:  PHASE 4: Post-processing 
  DEBUG onnx_ir::phases::post_processing: Starting Identity elimination
  DEBUG onnx_ir::phases::post_processing: Re-running constant lifting after Identity elimination
  DEBUG onnx_ir::pipeline:  PHASE 5: Finalization 
  DEBUG onnx_ir::pipeline:  PHASE 6: Node Conversion (RawNode -> Node) 
  DEBUG onnx_ir::ir::graph: Converting node 'constant2' of type Constant
  DEBUG onnx_ir::ir::graph: Converting node 'constant3' of type Constant
  DEBUG onnx_ir::ir::graph: Converting node 'constant4' of type Constant
  DEBUG onnx_ir::ir::graph: Converting node 'constant7' of type Constant
  DEBUG onnx_ir::ir::graph: Converting node 'constant8' of type Constant
  DEBUG onnx_ir::ir::graph: Converting node 'constant11' of type Constant
  DEBUG onnx_ir::ir::graph: Converting node 'constant12' of type Constant
  DEBUG onnx_ir::ir::graph: Converting node 'constant17' of type Constant
  DEBUG onnx_ir::ir::graph: Converting node 'constant18' of type Constant
  DEBUG onnx_ir::ir::graph: Converting node 'constant21' of type Constant
  DEBUG onnx_ir::ir::graph: Converting node 'constant22' of type Constant
  DEBUG onnx_ir::ir::graph: Converting node 'constant140' of type Constant
  DEBUG onnx_ir::ir::graph: Converting node 'constant141' of type Constant
  DEBUG onnx_ir::ir::graph: Converting node 'reducemin1' of type ReduceMin
  DEBUG onnx_ir::ir::graph: Converting node 'sub1' of type Sub
  DEBUG onnx_ir::ir::graph: Converting node 'reducemax1' of type ReduceMax
  DEBUG onnx_ir::ir::graph: Converting node 'add1' of type Add
  DEBUG onnx_ir::ir::graph: Converting node 'div1' of type Div
  DEBUG onnx_ir::ir::graph: Converting node 'sub2' of type Sub
  DEBUG onnx_ir::ir::graph: Converting node 'mul1' of type Mul
  DEBUG onnx_ir::ir::graph: Converting node 'reshape1' of type Reshape
  DEBUG onnx_ir::ir::graph: Converting node 'reshape2' of type Reshape
  DEBUG onnx_ir::ir::graph: Converting node 'gather1' of type Gather
  DEBUG onnx_ir::ir::graph: Converting node 'gather2' of type Gather
  DEBUG onnx_ir::ir::graph: Converting node 'reshape3' of type Reshape
  DEBUG onnx_ir::ir::graph: Converting node 'reshape4' of type Reshape
  DEBUG onnx_ir::ir::graph: Converting node 'mul2' of type Mul
  DEBUG onnx_ir::ir::graph: Converting node 'mul3' of type Mul
  DEBUG onnx_ir::ir::graph: Converting node 'linear1' of type Linear
  DEBUG onnx_ir::ir::graph: Converting node 'linear2' of type Linear
  DEBUG onnx_ir::ir::graph: Converting node 'linear3' of type Linear
  DEBUG onnx_ir::ir::graph: Converting node 'linear4' of type Linear
  DEBUG onnx_ir::ir::graph: Converting node 'mul4' of type Mul
  DEBUG onnx_ir::ir::graph: Converting node 'mul5' of type Mul
  DEBUG onnx_ir::ir::graph: Converting node 'pow1' of type Pow
  DEBUG onnx_ir::ir::graph: Converting node 'pow2' of type Pow
  DEBUG onnx_ir::ir::graph: Converting node 'slice1' of type Slice
  DEBUG onnx_ir::ir::graph: Converting node 'slice2' of type Slice
  DEBUG onnx_ir::ir::graph: Converting node 'unsqueeze1' of type Unsqueeze
  DEBUG onnx_ir::ir::graph: Converting node 'unsqueeze2' of type Unsqueeze
  DEBUG onnx_ir::ir::graph: Converting node 'concat1' of type Concat
  DEBUG onnx_ir::ir::graph: Converting node 'mul6' of type Mul
  DEBUG onnx_ir::ir::graph: Converting node 'add2' of type Add
  DEBUG onnx_ir::ir::graph: Converting node 'transpose1' of type Transpose
  DEBUG onnx_ir::ir::graph: Converting node 'conv2d1' of type Conv2d
  ERROR burn_import::logger: PANIC => panicked at /Users/justinc/.cargo/git/checkouts/burn-6c277d792b0d5d7a/439a26c/crates/onnx-ir/src/node/padding.rs:136:9:
  Asymmetric padding is not supported

  --- stderr

  thread 'main' (11321025) panicked at /Users/justinc/.cargo/git/checkouts/burn-6c277d792b0d5d7a/439a26c/crates/onnx-ir/src/node/padding.rs:136:9:
  Asymmetric padding is not supported
  stack backtrace:
     0: __rustc::rust_begin_unwind
               at /rustc/ded5c06cf21d2b93bffd5d884aa6e96934ee4234/library/std/src/panicking.rs:698:5
     1: core::panicking::panic_fmt
               at /rustc/ded5c06cf21d2b93bffd5d884aa6e96934ee4234/library/core/src/panicking.rs:80:14
     2: onnx_ir::node::padding::padding_config_2d
     3: <onnx_ir::node::conv2d::Conv2dProcessor as onnx_ir::processor::NodeProcessor>::extract_config
     4: <onnx_ir::node::conv2d::Conv2dProcessor as onnx_ir::processor::NodeProcessor>::build_node
     5: <T as onnx_ir::registry::ProcessorMethods>::build_node
     6: onnx_ir::ir::graph::convert_builders_to_nodes::{{closure}}
     7: core::iter::adapters::map::map_fold::{{closure}}
     8: <alloc::vec::into_iter::IntoIter<T,A> as core::iter::traits::iterator::Iterator>::fold
     9: <core::iter::adapters::map::Map<I,F> as core::iter::traits::iterator::Iterator>::fold
    10: core::iter::traits::iterator::Iterator::for_each
    11: alloc::vec::Vec<T,A>::extend_trusted
    12: <alloc::vec::Vec<T,A> as alloc::vec::spec_extend::SpecExtend<T,I>>::spec_extend
    13: <alloc::vec::Vec<T> as alloc::vec::spec_from_iter_nested::SpecFromIterNested<T,I>>::from_iter
    14: alloc::vec::in_place_collect::<impl alloc::vec::spec_from_iter::SpecFromIter<T,I> for alloc::vec::Vec<T>>::from_iter
    15: <alloc::vec::Vec<T> as core::iter::traits::collect::FromIterator<T>>::from_iter
    16: core::iter::traits::iterator::Iterator::collect
    17: onnx_ir::ir::graph::convert_builders_to_nodes
    18: onnx_ir::ir::graph::OnnxGraphBuilder::convert_to_graph
    19: onnx_ir::pipeline::build_graph_from_proto_with_base_path
    20: onnx_ir::pipeline::build_graph_with_base_path
    21: onnx_ir::pipeline::OnnxGraphBuilder::parse_buffer
    22: onnx_ir::pipeline::OnnxGraphBuilder::parse_file
    23: burn_import::onnx::model_gen::ModelGen::generate_model
    24: burn_import::onnx::model_gen::ModelGen::run
    25: burn_import::onnx::model_gen::ModelGen::run_from_script
    26: build_script_build::main
    27: core::ops::function::FnOnce::call_once
  note: Some details are omitted, run with `RUST_BACKTRACE=full` for a verbose backtrace.

This is my cargo.toml:

[package]
name = "my_burn_app"
version = "0.1.0"
edition = "2024"

# See more keys and their definitions at https://doc.rust-lang.org/cargo/reference/manifest.html
[build-dependencies]
burn-import = { git = "https://github.com/tracel-ai/burn.git" }

[dependencies]

burn = { git = "https://github.com/tracel-ai/burn.git" }

[justinchuby]

Also, I think the paddings are actually 0.

[antimora]

Also, I think the paddings are actually 0.

Thanks for reporting, will take a look.

[antimora]

We can probably fix this issue using the new Burn's Pad capability introduced in #4105

[justinchuby]

Looks like the node is this:

[antimora]

Looks like the node is this:

Thanks!

Yes, we can fix this by adding pad operations in generated code or updating Burn APIs

[antimora]

Fix submitted: #4263

[justinchuby]

@antimora

Thanks - I am now able to compile the app. However I get the error

thread 'main' (16654574) panicked at /Users/justinc/.cargo/git/checkouts/burn-c0f80f51d1aff506/b75d84c/crates/burn-backend/src/backend/ops/modules/conv.rs:149:6:
attempt to subtract with overflow
stack backtrace:
   0: __rustc::rust_begin_unwind
             at /rustc/ded5c06cf21d2b93bffd5d884aa6e96934ee4234/library/std/src/panicking.rs:698:5
   1: core::panicking::panic_fmt
             at /rustc/ded5c06cf21d2b93bffd5d884aa6e96934ee4234/library/core/src/panicking.rs:80:14
   2: core::panicking::panic_const::panic_const_sub_overflow
             at /rustc/ded5c06cf21d2b93bffd5d884aa6e96934ee4234/library/core/src/panicking.rs:180:17
   3: burn_backend::backend::ops::modules::conv::calculate_conv_output_size
             at /Users/justinc/.cargo/git/checkouts/burn-c0f80f51d1aff506/b75d84c/crates/burn-backend/src/backend/ops/modules/conv.rs:149:6
   4: burn_ndarray::ops::simd::conv::conv2d
             at /Users/justinc/.cargo/git/checkouts/burn-c0f80f51d1aff506/b75d84c/crates/burn-ndarray/src/ops/simd/conv.rs:80:22
   5: burn_ndarray::ops::simd::conv::try_conv2d_simd
             at /Users/justinc/.cargo/git/checkouts/burn-c0f80f51d1aff506/b75d84c/crates/burn-ndarray/src/ops/simd/conv.rs:27:23
   6: burn_ndarray::ops::module::<impl burn_backend::backend::ops::modules::base::ModuleOps<burn_ndarray::backend::NdArray<E,I,Q>> for burn_ndarray::backend::NdArray<E,I,Q>>::conv2d::{{closure}}
             at /Users/justinc/.cargo/git/checkouts/burn-c0f80f51d1aff506/b75d84c/crates/burn-ndarray/src/ops/module.rs:62:43
   7: burn_ndarray::ops::module::<impl burn_backend::backend::ops::modules::base::ModuleOps<burn_ndarray::backend::NdArray<E,I,Q>> for burn_ndarray::backend::NdArray<E,I,Q>>::conv2d
             at /Users/justinc/.cargo/git/checkouts/burn-c0f80f51d1aff506/b75d84c/crates/burn-ndarray/src/ops/module.rs:31:17
   8: burn_tensor::tensor::module::conv2d
             at /Users/justinc/.cargo/git/checkouts/burn-c0f80f51d1aff506/b75d84c/crates/burn-tensor/src/tensor/module.rs:62:40
   9: burn_nn::modules::conv::conv2d::Conv2d<B>::forward
             at /Users/justinc/.cargo/git/checkouts/burn-c0f80f51d1aff506/b75d84c/crates/burn-nn/src/modules/conv/conv2d.rs:177:13
  10: my_burn_app::model::my_model::Model<B>::forward
             at ./target/debug/build/my_burn_app-30fc87b67c1fc028/out/model/new_birdnet.rs:898:41
  11: my_burn_app::main
             at ./src/main.rs:24:24
  12: core::ops::function::FnOnce::call_once
             at /Users/justinc/.rustup/toolchains/stable-aarch64-apple-darwin/lib/rustlib/src/rust/library/core/src/ops/function.rs:250:5

when running it.

This is the built model

// Generated from ONNX "/Users/justinc/Documents/GitHub/my_burn_app/new_birdnet.onnx" by burn-import
use burn::prelude::*;
use burn::nn::Linear;
use burn::nn::LinearConfig;
use burn::nn::PaddingConfig2d;
use burn::nn::conv::Conv2d;
use burn::nn::conv::Conv2dConfig;
use burn::nn::pool::AvgPool2d;
use burn::nn::pool::AvgPool2dConfig;
use burn::nn::pool::MaxPool2d;
use burn::nn::pool::MaxPool2dConfig;
use burn_store::BurnpackStore;
use burn_store::ModuleSnapshot;


#[derive(Module, Debug)]
pub struct Model<B: Backend> {
    constant7: burn::module::Param<Tensor<B, 2, Int>>,
    constant8: burn::module::Param<Tensor<B, 2, Int>>,
    constant11: burn::module::Param<Tensor<B, 1>>,
    constant12: burn::module::Param<Tensor<B, 1>>,
    constant21: burn::module::Param<Tensor<B, 4>>,
    constant22: burn::module::Param<Tensor<B, 4>>,
    constant140: burn::module::Param<Tensor<B, 4>>,
    constant141: burn::module::Param<Tensor<B, 4>>,
    linear1: Linear<B>,
    linear2: Linear<B>,
    linear3: Linear<B>,
    linear4: Linear<B>,
    conv2d1: Conv2d<B>,
    maxpool2d1: MaxPool2d,
    averagepool2d1: AvgPool2d,
    conv2d2: Conv2d<B>,
    conv2d3: Conv2d<B>,
    conv2d4: Conv2d<B>,
    conv2d5: Conv2d<B>,
    conv2d6: Conv2d<B>,
    conv2d7: Conv2d<B>,
    conv2d8: Conv2d<B>,
    conv2d9: Conv2d<B>,
    conv2d10: Conv2d<B>,
    conv2d11: Conv2d<B>,
    conv2d12: Conv2d<B>,
    conv2d13: Conv2d<B>,
    conv2d14: Conv2d<B>,
    conv2d15: Conv2d<B>,
    conv2d16: Conv2d<B>,
    conv2d17: Conv2d<B>,
    conv2d18: Conv2d<B>,
    conv2d19: Conv2d<B>,
    conv2d20: Conv2d<B>,
    conv2d21: Conv2d<B>,
    conv2d22: Conv2d<B>,
    conv2d23: Conv2d<B>,
    conv2d24: Conv2d<B>,
    conv2d25: Conv2d<B>,
    conv2d26: Conv2d<B>,
    conv2d27: Conv2d<B>,
    conv2d28: Conv2d<B>,
    conv2d29: Conv2d<B>,
    conv2d30: Conv2d<B>,
    conv2d31: Conv2d<B>,
    conv2d32: Conv2d<B>,
    conv2d33: Conv2d<B>,
    conv2d34: Conv2d<B>,
    conv2d35: Conv2d<B>,
    conv2d36: Conv2d<B>,
    conv2d37: Conv2d<B>,
    conv2d38: Conv2d<B>,
    conv2d39: Conv2d<B>,
    conv2d40: Conv2d<B>,
    conv2d41: Conv2d<B>,
    conv2d42: Conv2d<B>,
    conv2d43: Conv2d<B>,
    conv2d44: Conv2d<B>,
    conv2d45: Conv2d<B>,
    conv2d46: Conv2d<B>,
    conv2d47: Conv2d<B>,
    conv2d48: Conv2d<B>,
    conv2d49: Conv2d<B>,
    conv2d50: Conv2d<B>,
    conv2d51: Conv2d<B>,
    conv2d52: Conv2d<B>,
    conv2d53: Conv2d<B>,
    conv2d54: Conv2d<B>,
    conv2d55: Conv2d<B>,
    conv2d56: Conv2d<B>,
    conv2d57: Conv2d<B>,
    conv2d58: Conv2d<B>,
    conv2d59: Conv2d<B>,
    conv2d60: Conv2d<B>,
    conv2d61: Conv2d<B>,
    conv2d62: Conv2d<B>,
    conv2d63: Conv2d<B>,
    conv2d64: Conv2d<B>,
    conv2d65: Conv2d<B>,
    conv2d66: Conv2d<B>,
    conv2d67: Conv2d<B>,
    conv2d68: Conv2d<B>,
    conv2d69: Conv2d<B>,
    conv2d70: Conv2d<B>,
    conv2d71: Conv2d<B>,
    conv2d72: Conv2d<B>,
    conv2d73: Conv2d<B>,
    conv2d74: Conv2d<B>,
    conv2d75: Conv2d<B>,
    conv2d76: Conv2d<B>,
    conv2d77: Conv2d<B>,
    linear5: Linear<B>,
    phantom: core::marker::PhantomData<B>,
    device: burn::module::Ignored<B::Device>,
}


impl<B: Backend> Default for Model<B> {
    fn default() -> Self {
        Self::from_file(
            "/Users/justinc/Documents/GitHub/my_burn_app/target/debug/build/my_burn_app-30fc87b67c1fc028/out/model/new_birdnet.bpk",
            &Default::default(),
        )
    }
}

impl<B: Backend> Model<B> {
    /// Load model weights from a burnpack file.
    pub fn from_file(file: &str, device: &B::Device) -> Self {
        let mut model = Self::new(device);
        let mut store = BurnpackStore::from_file(file);
        model.load_from(&mut store).expect("Failed to load burnpack file");
        model
    }
}

impl<B: Backend> Model<B> {
    #[allow(unused_variables)]
    pub fn new(device: &B::Device) -> Self {
        let constant7: burn::module::Param<Tensor<B, 2, Int>> = burn::module::Param::uninitialized(
            burn::module::ParamId::new(),
            move |device, _require_grad| Tensor::<B, 2, Int>::zeros([511, 1024], device),
            device.clone(),
            false,
            [511, 1024].into(),
        );
        let constant8: burn::module::Param<Tensor<B, 2, Int>> = burn::module::Param::uninitialized(
            burn::module::ParamId::new(),
            move |device, _require_grad| Tensor::<B, 2, Int>::zeros([511, 128], device),
            device.clone(),
            false,
            [511, 128].into(),
        );
        let constant11: burn::module::Param<Tensor<B, 1>> = burn::module::Param::uninitialized(
            burn::module::ParamId::new(),
            move |device, _require_grad| Tensor::<B, 1>::zeros([2048], device),
            device.clone(),
            false,
            [2048].into(),
        );
        let constant12: burn::module::Param<Tensor<B, 1>> = burn::module::Param::uninitialized(
            burn::module::ParamId::new(),
            move |device, _require_grad| Tensor::<B, 1>::zeros([1024], device),
            device.clone(),
            false,
            [1024].into(),
        );
        let constant21: burn::module::Param<Tensor<B, 4>> = burn::module::Param::uninitialized(
            burn::module::ParamId::new(),
            move |device, _require_grad| Tensor::<B, 4>::zeros([1, 1, 1, 2], device),
            device.clone(),
            false,
            [1, 1, 1, 2].into(),
        );
        let constant22: burn::module::Param<Tensor<B, 4>> = burn::module::Param::uninitialized(
            burn::module::ParamId::new(),
            move |device, _require_grad| Tensor::<B, 4>::zeros([1, 1, 1, 2], device),
            device.clone(),
            false,
            [1, 1, 1, 2].into(),
        );
        let constant140: burn::module::Param<Tensor<B, 4>> = burn::module::Param::uninitialized(
            burn::module::ParamId::new(),
            move |device, _require_grad| Tensor::<B, 4>::zeros([1, 192, 1, 1], device),
            device.clone(),
            false,
            [1, 192, 1, 1].into(),
        );
        let constant141: burn::module::Param<Tensor<B, 4>> = burn::module::Param::uninitialized(
            burn::module::ParamId::new(),
            move |device, _require_grad| Tensor::<B, 4>::zeros([1, 192, 1, 1], device),
            device.clone(),
            false,
            [1, 192, 1, 1].into(),
        );
        let linear1 = LinearConfig::new(2048, 1025).with_bias(false).init(device);
        let linear2 = LinearConfig::new(1024, 513).with_bias(false).init(device);
        let linear3 = LinearConfig::new(1025, 96).with_bias(false).init(device);
        let linear4 = LinearConfig::new(513, 96).with_bias(false).init(device);
        let conv2d1 = Conv2dConfig::new([2, 24], [4, 8])
            .with_stride([2, 2])
            .with_padding(PaddingConfig2d::Explicit(1, 3, 1, 4))
            .with_dilation([1, 1])
            .with_groups(1)
            .with_bias(true)
            .init(device);
        let maxpool2d1 = MaxPool2dConfig::new([1, 2])
            .with_strides([1, 2])
            .with_padding(PaddingConfig2d::Valid)
            .with_dilation([1, 1])
            .with_ceil_mode(false)
            .init();
        let averagepool2d1 = AvgPool2dConfig::new([1, 2])
            .with_strides([1, 2])
            .with_padding(PaddingConfig2d::Valid)
            .with_count_include_pad(false)
            .with_ceil_mode(false)
            .init();
        let conv2d2 = Conv2dConfig::new([48, 24], [1, 1])
            .with_stride([1, 1])
            .with_padding(PaddingConfig2d::Valid)
            .with_dilation([1, 1])
            .with_groups(1)
            .with_bias(true)
            .init(device);
        let conv2d3 = Conv2dConfig::new([24, 72], [1, 1])
            .with_stride([1, 1])
            .with_padding(PaddingConfig2d::Valid)
            .with_dilation([1, 1])
            .with_groups(1)
            .with_bias(true)
            .init(device);
        let conv2d4 = Conv2dConfig::new([72, 72], [3, 3])
            .with_stride([2, 2])
            .with_padding(PaddingConfig2d::Explicit(1, 1, 1, 1))
            .with_dilation([1, 1])
            .with_groups(72)
            .with_bias(true)
            .init(device);
        let conv2d5 = Conv2dConfig::new([72, 36], [1, 1])
            .with_stride([1, 1])
            .with_padding(PaddingConfig2d::Valid)
            .with_dilation([1, 1])
            .with_groups(1)
            .with_bias(false)
            .init(device);
        let conv2d6 = Conv2dConfig::new([36, 72], [1, 1])
            .with_stride([1, 1])
            .with_padding(PaddingConfig2d::Valid)
            .with_dilation([1, 1])
            .with_groups(1)
            .with_bias(true)
            .init(device);
        let conv2d7 = Conv2dConfig::new([72, 72], [3, 3])
            .with_stride([1, 1])
            .with_padding(PaddingConfig2d::Explicit(1, 1, 1, 1))
            .with_dilation([1, 1])
            .with_groups(72)
            .with_bias(true)
            .init(device);
        let conv2d8 = Conv2dConfig::new([72, 36], [1, 1])
            .with_stride([1, 1])
            .with_padding(PaddingConfig2d::Valid)
            .with_dilation([1, 1])
            .with_groups(1)
            .with_bias(false)
            .init(device);
        let conv2d9 = Conv2dConfig::new([36, 72], [1, 1])
            .with_stride([1, 1])
            .with_padding(PaddingConfig2d::Valid)
            .with_dilation([1, 1])
            .with_groups(1)
            .with_bias(true)
            .init(device);
        let conv2d10 = Conv2dConfig::new([72, 72], [3, 3])
            .with_stride([1, 1])
            .with_padding(PaddingConfig2d::Explicit(1, 1, 1, 1))
            .with_dilation([1, 1])
            .with_groups(72)
            .with_bias(true)
            .init(device);
        let conv2d11 = Conv2dConfig::new([72, 36], [1, 1])
            .with_stride([1, 1])
            .with_padding(PaddingConfig2d::Valid)
            .with_dilation([1, 1])
            .with_groups(1)
            .with_bias(false)
            .init(device);
        let conv2d12 = Conv2dConfig::new([36, 288], [1, 1])
            .with_stride([1, 1])
            .with_padding(PaddingConfig2d::Valid)
            .with_dilation([1, 1])
            .with_groups(1)
            .with_bias(true)
            .init(device);
        let conv2d13 = Conv2dConfig::new([288, 288], [3, 3])
            .with_stride([2, 2])
            .with_padding(PaddingConfig2d::Explicit(1, 1, 1, 1))
            .with_dilation([1, 1])
            .with_groups(288)
            .with_bias(true)
            .init(device);
        let conv2d14 = Conv2dConfig::new([288, 18], [1, 1])
            .with_stride([1, 1])
            .with_padding(PaddingConfig2d::Valid)
            .with_dilation([1, 1])
            .with_groups(1)
            .with_bias(false)
            .init(device);
        let conv2d15 = Conv2dConfig::new([18, 288], [1, 1])
            .with_stride([1, 1])
            .with_padding(PaddingConfig2d::Valid)
            .with_dilation([1, 1])
            .with_groups(1)
            .with_bias(false)
            .init(device);
        let conv2d16 = Conv2dConfig::new([288, 72], [1, 1])
            .with_stride([1, 1])
            .with_padding(PaddingConfig2d::Valid)
            .with_dilation([1, 1])
            .with_groups(1)
            .with_bias(false)
            .init(device);
        let conv2d17 = Conv2dConfig::new([72, 288], [1, 1])
            .with_stride([1, 1])
            .with_padding(PaddingConfig2d::Valid)
            .with_dilation([1, 1])
            .with_groups(1)
            .with_bias(true)
            .init(device);
        let conv2d18 = Conv2dConfig::new([288, 288], [3, 3])
            .with_stride([1, 1])
            .with_padding(PaddingConfig2d::Explicit(1, 1, 1, 1))
            .with_dilation([1, 1])
            .with_groups(288)
            .with_bias(true)
            .init(device);
        let conv2d19 = Conv2dConfig::new([288, 18], [1, 1])
            .with_stride([1, 1])
            .with_padding(PaddingConfig2d::Valid)
            .with_dilation([1, 1])
            .with_groups(1)
            .with_bias(false)
            .init(device);
        let conv2d20 = Conv2dConfig::new([18, 288], [1, 1])
            .with_stride([1, 1])
            .with_padding(PaddingConfig2d::Valid)
            .with_dilation([1, 1])
            .with_groups(1)
            .with_bias(false)
            .init(device);
        let conv2d21 = Conv2dConfig::new([288, 72], [1, 1])
            .with_stride([1, 1])
            .with_padding(PaddingConfig2d::Valid)
            .with_dilation([1, 1])
            .with_groups(1)
            .with_bias(false)
            .init(device);
        let conv2d22 = Conv2dConfig::new([72, 288], [1, 1])
            .with_stride([1, 1])
            .with_padding(PaddingConfig2d::Valid)
            .with_dilation([1, 1])
            .with_groups(1)
            .with_bias(true)
            .init(device);
        let conv2d23 = Conv2dConfig::new([288, 288], [3, 3])
            .with_stride([1, 1])
            .with_padding(PaddingConfig2d::Explicit(1, 1, 1, 1))
            .with_dilation([1, 1])
            .with_groups(288)
            .with_bias(true)
            .init(device);
        let conv2d24 = Conv2dConfig::new([288, 18], [1, 1])
            .with_stride([1, 1])
            .with_padding(PaddingConfig2d::Valid)
            .with_dilation([1, 1])
            .with_groups(1)
            .with_bias(false)
            .init(device);
        let conv2d25 = Conv2dConfig::new([18, 288], [1, 1])
            .with_stride([1, 1])
            .with_padding(PaddingConfig2d::Valid)
            .with_dilation([1, 1])
            .with_groups(1)
            .with_bias(false)
            .init(device);
        let conv2d26 = Conv2dConfig::new([288, 72], [1, 1])
            .with_stride([1, 1])
            .with_padding(PaddingConfig2d::Valid)
            .with_dilation([1, 1])
            .with_groups(1)
            .with_bias(false)
            .init(device);
        let conv2d27 = Conv2dConfig::new([72, 288], [1, 1])
            .with_stride([1, 1])
            .with_padding(PaddingConfig2d::Valid)
            .with_dilation([1, 1])
            .with_groups(1)
            .with_bias(true)
            .init(device);
        let conv2d28 = Conv2dConfig::new([288, 288], [3, 3])
            .with_stride([1, 1])
            .with_padding(PaddingConfig2d::Explicit(1, 1, 1, 1))
            .with_dilation([1, 1])
            .with_groups(288)
            .with_bias(true)
            .init(device);
        let conv2d29 = Conv2dConfig::new([288, 18], [1, 1])
            .with_stride([1, 1])
            .with_padding(PaddingConfig2d::Valid)
            .with_dilation([1, 1])
            .with_groups(1)
            .with_bias(false)
            .init(device);
        let conv2d30 = Conv2dConfig::new([18, 288], [1, 1])
            .with_stride([1, 1])
            .with_padding(PaddingConfig2d::Valid)
            .with_dilation([1, 1])
            .with_groups(1)
            .with_bias(false)
            .init(device);
        let conv2d31 = Conv2dConfig::new([288, 72], [1, 1])
            .with_stride([1, 1])
            .with_padding(PaddingConfig2d::Valid)
            .with_dilation([1, 1])
            .with_groups(1)
            .with_bias(false)
            .init(device);
        let conv2d32 = Conv2dConfig::new([72, 864], [1, 1])
            .with_stride([1, 1])
            .with_padding(PaddingConfig2d::Valid)
            .with_dilation([1, 1])
            .with_groups(1)
            .with_bias(true)
            .init(device);
        let conv2d33 = Conv2dConfig::new([864, 864], [3, 3])
            .with_stride([2, 2])
            .with_padding(PaddingConfig2d::Explicit(1, 1, 1, 1))
            .with_dilation([1, 1])
            .with_groups(864)
            .with_bias(true)
            .init(device);
        let conv2d34 = Conv2dConfig::new([864, 27], [1, 1])
            .with_stride([1, 1])
            .with_padding(PaddingConfig2d::Valid)
            .with_dilation([1, 1])
            .with_groups(1)
            .with_bias(false)
            .init(device);
        let conv2d35 = Conv2dConfig::new([27, 864], [1, 1])
            .with_stride([1, 1])
            .with_padding(PaddingConfig2d::Valid)
            .with_dilation([1, 1])
            .with_groups(1)
            .with_bias(false)
            .init(device);
        let conv2d36 = Conv2dConfig::new([864, 108], [1, 1])
            .with_stride([1, 1])
            .with_padding(PaddingConfig2d::Valid)
            .with_dilation([1, 1])
            .with_groups(1)
            .with_bias(false)
            .init(device);
        let conv2d37 = Conv2dConfig::new([108, 864], [1, 1])
            .with_stride([1, 1])
            .with_padding(PaddingConfig2d::Valid)
            .with_dilation([1, 1])
            .with_groups(1)
            .with_bias(true)
            .init(device);
        let conv2d38 = Conv2dConfig::new([864, 864], [3, 3])
            .with_stride([1, 1])
            .with_padding(PaddingConfig2d::Explicit(1, 1, 1, 1))
            .with_dilation([1, 1])
            .with_groups(864)
            .with_bias(true)
            .init(device);
        let conv2d39 = Conv2dConfig::new([864, 27], [1, 1])
            .with_stride([1, 1])
            .with_padding(PaddingConfig2d::Valid)
            .with_dilation([1, 1])
            .with_groups(1)
            .with_bias(false)
            .init(device);
        let conv2d40 = Conv2dConfig::new([27, 864], [1, 1])
            .with_stride([1, 1])
            .with_padding(PaddingConfig2d::Valid)
            .with_dilation([1, 1])
            .with_groups(1)
            .with_bias(false)
            .init(device);
        let conv2d41 = Conv2dConfig::new([864, 108], [1, 1])
            .with_stride([1, 1])
            .with_padding(PaddingConfig2d::Valid)
            .with_dilation([1, 1])
            .with_groups(1)
            .with_bias(false)
            .init(device);
        let conv2d42 = Conv2dConfig::new([108, 864], [1, 1])
            .with_stride([1, 1])
            .with_padding(PaddingConfig2d::Valid)
            .with_dilation([1, 1])
            .with_groups(1)
            .with_bias(true)
            .init(device);
        let conv2d43 = Conv2dConfig::new([864, 864], [3, 3])
            .with_stride([1, 1])
            .with_padding(PaddingConfig2d::Explicit(1, 1, 1, 1))
            .with_dilation([1, 1])
            .with_groups(864)
            .with_bias(true)
            .init(device);
        let conv2d44 = Conv2dConfig::new([864, 27], [1, 1])
            .with_stride([1, 1])
            .with_padding(PaddingConfig2d::Valid)
            .with_dilation([1, 1])
            .with_groups(1)
            .with_bias(false)
            .init(device);
        let conv2d45 = Conv2dConfig::new([27, 864], [1, 1])
            .with_stride([1, 1])
            .with_padding(PaddingConfig2d::Valid)
            .with_dilation([1, 1])
            .with_groups(1)
            .with_bias(false)
            .init(device);
        let conv2d46 = Conv2dConfig::new([864, 108], [1, 1])
            .with_stride([1, 1])
            .with_padding(PaddingConfig2d::Valid)
            .with_dilation([1, 1])
            .with_groups(1)
            .with_bias(false)
            .init(device);
        let conv2d47 = Conv2dConfig::new([108, 864], [1, 1])
            .with_stride([1, 1])
            .with_padding(PaddingConfig2d::Valid)
            .with_dilation([1, 1])
            .with_groups(1)
            .with_bias(true)
            .init(device);
        let conv2d48 = Conv2dConfig::new([864, 864], [3, 3])
            .with_stride([1, 1])
            .with_padding(PaddingConfig2d::Explicit(1, 1, 1, 1))
            .with_dilation([1, 1])
            .with_groups(864)
            .with_bias(true)
            .init(device);
        let conv2d49 = Conv2dConfig::new([864, 27], [1, 1])
            .with_stride([1, 1])
            .with_padding(PaddingConfig2d::Valid)
            .with_dilation([1, 1])
            .with_groups(1)
            .with_bias(false)
            .init(device);
        let conv2d50 = Conv2dConfig::new([27, 864], [1, 1])
            .with_stride([1, 1])
            .with_padding(PaddingConfig2d::Valid)
            .with_dilation([1, 1])
            .with_groups(1)
            .with_bias(false)
            .init(device);
        let conv2d51 = Conv2dConfig::new([864, 108], [1, 1])
            .with_stride([1, 1])
            .with_padding(PaddingConfig2d::Valid)
            .with_dilation([1, 1])
            .with_groups(1)
            .with_bias(false)
            .init(device);
        let conv2d52 = Conv2dConfig::new([108, 864], [1, 1])
            .with_stride([1, 1])
            .with_padding(PaddingConfig2d::Valid)
            .with_dilation([1, 1])
            .with_groups(1)
            .with_bias(true)
            .init(device);
        let conv2d53 = Conv2dConfig::new([864, 864], [3, 3])
            .with_stride([1, 1])
            .with_padding(PaddingConfig2d::Explicit(1, 1, 1, 1))
            .with_dilation([1, 1])
            .with_groups(864)
            .with_bias(true)
            .init(device);
        let conv2d54 = Conv2dConfig::new([864, 27], [1, 1])
            .with_stride([1, 1])
            .with_padding(PaddingConfig2d::Valid)
            .with_dilation([1, 1])
            .with_groups(1)
            .with_bias(false)
            .init(device);
        let conv2d55 = Conv2dConfig::new([27, 864], [1, 1])
            .with_stride([1, 1])
            .with_padding(PaddingConfig2d::Valid)
            .with_dilation([1, 1])
            .with_groups(1)
            .with_bias(false)
            .init(device);
        let conv2d56 = Conv2dConfig::new([864, 108], [1, 1])
            .with_stride([1, 1])
            .with_padding(PaddingConfig2d::Valid)
            .with_dilation([1, 1])
            .with_groups(1)
            .with_bias(false)
            .init(device);
        let conv2d57 = Conv2dConfig::new([108, 1536], [1, 1])
            .with_stride([1, 1])
            .with_padding(PaddingConfig2d::Valid)
            .with_dilation([1, 1])
            .with_groups(1)
            .with_bias(true)
            .init(device);
        let conv2d58 = Conv2dConfig::new([1536, 1536], [3, 3])
            .with_stride([2, 2])
            .with_padding(PaddingConfig2d::Explicit(1, 1, 1, 1))
            .with_dilation([1, 1])
            .with_groups(1536)
            .with_bias(true)
            .init(device);
        let conv2d59 = Conv2dConfig::new([1536, 48], [1, 1])
            .with_stride([1, 1])
            .with_padding(PaddingConfig2d::Valid)
            .with_dilation([1, 1])
            .with_groups(1)
            .with_bias(false)
            .init(device);
        let conv2d60 = Conv2dConfig::new([48, 1536], [1, 1])
            .with_stride([1, 1])
            .with_padding(PaddingConfig2d::Valid)
            .with_dilation([1, 1])
            .with_groups(1)
            .with_bias(false)
            .init(device);
        let conv2d61 = Conv2dConfig::new([1536, 192], [1, 1])
            .with_stride([1, 1])
            .with_padding(PaddingConfig2d::Valid)
            .with_dilation([1, 1])
            .with_groups(1)
            .with_bias(false)
            .init(device);
        let conv2d62 = Conv2dConfig::new([192, 1536], [1, 1])
            .with_stride([1, 1])
            .with_padding(PaddingConfig2d::Valid)
            .with_dilation([1, 1])
            .with_groups(1)
            .with_bias(true)
            .init(device);
        let conv2d63 = Conv2dConfig::new([1536, 1536], [3, 3])
            .with_stride([1, 1])
            .with_padding(PaddingConfig2d::Explicit(1, 1, 1, 1))
            .with_dilation([1, 1])
            .with_groups(1536)
            .with_bias(true)
            .init(device);
        let conv2d64 = Conv2dConfig::new([1536, 48], [1, 1])
            .with_stride([1, 1])
            .with_padding(PaddingConfig2d::Valid)
            .with_dilation([1, 1])
            .with_groups(1)
            .with_bias(false)
            .init(device);
        let conv2d65 = Conv2dConfig::new([48, 1536], [1, 1])
            .with_stride([1, 1])
            .with_padding(PaddingConfig2d::Valid)
            .with_dilation([1, 1])
            .with_groups(1)
            .with_bias(false)
            .init(device);
        let conv2d66 = Conv2dConfig::new([1536, 192], [1, 1])
            .with_stride([1, 1])
            .with_padding(PaddingConfig2d::Valid)
            .with_dilation([1, 1])
            .with_groups(1)
            .with_bias(false)
            .init(device);
        let conv2d67 = Conv2dConfig::new([192, 1536], [1, 1])
            .with_stride([1, 1])
            .with_padding(PaddingConfig2d::Valid)
            .with_dilation([1, 1])
            .with_groups(1)
            .with_bias(true)
            .init(device);
        let conv2d68 = Conv2dConfig::new([1536, 1536], [3, 3])
            .with_stride([1, 1])
            .with_padding(PaddingConfig2d::Explicit(1, 1, 1, 1))
            .with_dilation([1, 1])
            .with_groups(1536)
            .with_bias(true)
            .init(device);
        let conv2d69 = Conv2dConfig::new([1536, 48], [1, 1])
            .with_stride([1, 1])
            .with_padding(PaddingConfig2d::Valid)
            .with_dilation([1, 1])
            .with_groups(1)
            .with_bias(false)
            .init(device);
        let conv2d70 = Conv2dConfig::new([48, 1536], [1, 1])
            .with_stride([1, 1])
            .with_padding(PaddingConfig2d::Valid)
            .with_dilation([1, 1])
            .with_groups(1)
            .with_bias(false)
            .init(device);
        let conv2d71 = Conv2dConfig::new([1536, 192], [1, 1])
            .with_stride([1, 1])
            .with_padding(PaddingConfig2d::Valid)
            .with_dilation([1, 1])
            .with_groups(1)
            .with_bias(false)
            .init(device);
        let conv2d72 = Conv2dConfig::new([192, 1536], [1, 1])
            .with_stride([1, 1])
            .with_padding(PaddingConfig2d::Valid)
            .with_dilation([1, 1])
            .with_groups(1)
            .with_bias(true)
            .init(device);
        let conv2d73 = Conv2dConfig::new([1536, 1536], [3, 3])
            .with_stride([1, 1])
            .with_padding(PaddingConfig2d::Explicit(1, 1, 1, 1))
            .with_dilation([1, 1])
            .with_groups(1536)
            .with_bias(true)
            .init(device);
        let conv2d74 = Conv2dConfig::new([1536, 48], [1, 1])
            .with_stride([1, 1])
            .with_padding(PaddingConfig2d::Valid)
            .with_dilation([1, 1])
            .with_groups(1)
            .with_bias(false)
            .init(device);
        let conv2d75 = Conv2dConfig::new([48, 1536], [1, 1])
            .with_stride([1, 1])
            .with_padding(PaddingConfig2d::Valid)
            .with_dilation([1, 1])
            .with_groups(1)
            .with_bias(false)
            .init(device);
        let conv2d76 = Conv2dConfig::new([1536, 192], [1, 1])
            .with_stride([1, 1])
            .with_padding(PaddingConfig2d::Valid)
            .with_dilation([1, 1])
            .with_groups(1)
            .with_bias(false)
            .init(device);
        let conv2d77 = Conv2dConfig::new([192, 1024], [3, 3])
            .with_stride([1, 1])
            .with_padding(PaddingConfig2d::Valid)
            .with_dilation([1, 1])
            .with_groups(1)
            .with_bias(true)
            .init(device);
        let linear5 = LinearConfig::new(1024, 6522).with_bias(true).init(device);
        Self {
            constant7,
            constant8,
            constant11,
            constant12,
            constant21,
            constant22,
            constant140,
            constant141,
            linear1,
            linear2,
            linear3,
            linear4,
            conv2d1,
            maxpool2d1,
            averagepool2d1,
            conv2d2,
            conv2d3,
            conv2d4,
            conv2d5,
            conv2d6,
            conv2d7,
            conv2d8,
            conv2d9,
            conv2d10,
            conv2d11,
            conv2d12,
            conv2d13,
            conv2d14,
            conv2d15,
            conv2d16,
            conv2d17,
            conv2d18,
            conv2d19,
            conv2d20,
            conv2d21,
            conv2d22,
            conv2d23,
            conv2d24,
            conv2d25,
            conv2d26,
            conv2d27,
            conv2d28,
            conv2d29,
            conv2d30,
            conv2d31,
            conv2d32,
            conv2d33,
            conv2d34,
            conv2d35,
            conv2d36,
            conv2d37,
            conv2d38,
            conv2d39,
            conv2d40,
            conv2d41,
            conv2d42,
            conv2d43,
            conv2d44,
            conv2d45,
            conv2d46,
            conv2d47,
            conv2d48,
            conv2d49,
            conv2d50,
            conv2d51,
            conv2d52,
            conv2d53,
            conv2d54,
            conv2d55,
            conv2d56,
            conv2d57,
            conv2d58,
            conv2d59,
            conv2d60,
            conv2d61,
            conv2d62,
            conv2d63,
            conv2d64,
            conv2d65,
            conv2d66,
            conv2d67,
            conv2d68,
            conv2d69,
            conv2d70,
            conv2d71,
            conv2d72,
            conv2d73,
            conv2d74,
            conv2d75,
            conv2d76,
            conv2d77,
            linear5,
            phantom: core::marker::PhantomData,
            device: burn::module::Ignored(device.clone()),
        }
    }

    #[allow(clippy::let_and_return, clippy::approx_constant)]
    pub fn forward(&self, input: Tensor<B, 2>) -> Tensor<B, 2> {
        let constant2_out1 = 0.000001f32;
        let constant3_out1 = 0.5f32;
        let constant4_out1 = 2f32;
        let constant7_out1 = self.constant7.val();
        let constant8_out1 = self.constant8.val();
        let constant11_out1 = self.constant11.val();
        let constant12_out1 = self.constant12.val();
        let constant17_out1 = 0.22952409f32;
        let constant18_out1 = 0.1905273f32;
        let constant21_out1 = self.constant21.val();
        let constant22_out1 = self.constant22.val();
        let constant140_out1 = self.constant140.val();
        let constant141_out1 = self.constant141.val();
        let reducemin1_out1 = { input.clone().min_dim(1usize) };
        let sub1_out1 = input.sub(reducemin1_out1);
        let reducemax1_out1 = { sub1_out1.clone().max_dim(1usize) };
        let add1_out1 = reducemax1_out1.add_scalar(constant2_out1);
        let div1_out1 = sub1_out1.div(add1_out1);
        let sub2_out1 = div1_out1.sub_scalar(constant3_out1);
        let mul1_out1 = sub2_out1.mul_scalar(constant4_out1);
        let reshape1_out1 = mul1_out1.clone().reshape([-1, 72000, 2]);
        let reshape2_out1 = mul1_out1.reshape([-1, 18000, 8]);
        let gather1_out1 = reshape1_out1.take::<2, 4>(1, constant7_out1);
        let gather2_out1 = reshape2_out1.take::<2, 4>(1, constant8_out1);
        let reshape3_out1 = gather1_out1.reshape([-1, 511, 2048]);
        let reshape4_out1 = gather2_out1.reshape([-1, 511, 1024]);
        let mul2_out1 = reshape3_out1
            .mul(constant11_out1.unsqueeze_dims(&[0isize, 1isize]));
        let mul3_out1 = reshape4_out1
            .mul(constant12_out1.unsqueeze_dims(&[0isize, 1isize]));
        let linear1_out1 = self.linear1.forward(mul2_out1);
        let linear2_out1 = self.linear2.forward(mul3_out1);
        let linear3_out1 = self.linear3.forward(linear1_out1);
        let linear4_out1 = self.linear4.forward(linear2_out1);
        let mul4_out1 = linear3_out1.clone().mul(linear3_out1);
        let mul5_out1 = linear4_out1.clone().mul(linear4_out1);
        let pow1_out1 = mul4_out1.powf_scalar(constant17_out1);
        let pow2_out1 = mul5_out1.powf_scalar(constant18_out1);
        let slice1_out1 = pow1_out1.slice(s![.., .., - 1.. - 1000; - 1]);
        let slice2_out1 = pow2_out1.slice(s![.., .., - 1.. - 1000; - 1]);
        let unsqueeze1_out1: Tensor<B, 4> = slice1_out1.unsqueeze_dims::<4>(&[-1]);
        let unsqueeze2_out1: Tensor<B, 4> = slice2_out1.unsqueeze_dims::<4>(&[-1]);
        let concat1_out1 = burn::tensor::Tensor::cat(
            [unsqueeze1_out1, unsqueeze2_out1].into(),
            3,
        );
        let mul6_out1 = concat1_out1.mul(constant21_out1);
        let add2_out1 = mul6_out1.add(constant22_out1);
        let transpose1_out1 = add2_out1.permute([0, 3, 2, 1]);
        let conv2d1_out1 = self.conv2d1.forward(transpose1_out1);
        let relu1_out1 = burn::tensor::activation::relu(conv2d1_out1);
        let maxpool2d1_out1 = self.maxpool2d1.forward(relu1_out1.clone());
        let averagepool2d1_out1 = self.averagepool2d1.forward(relu1_out1);
        let concat2_out1 = burn::tensor::Tensor::cat(
            [maxpool2d1_out1, averagepool2d1_out1].into(),
            1,
        );
        let conv2d2_out1 = self.conv2d2.forward(concat2_out1);
        let conv2d3_out1 = self.conv2d3.forward(conv2d2_out1);
        let sigmoid1_out1 = burn::tensor::activation::sigmoid(conv2d3_out1.clone());
        let mul7_out1 = conv2d3_out1.mul(sigmoid1_out1);
        let conv2d4_out1 = self.conv2d4.forward(mul7_out1);
        let sigmoid2_out1 = burn::tensor::activation::sigmoid(conv2d4_out1.clone());
        let mul8_out1 = conv2d4_out1.mul(sigmoid2_out1);
        let conv2d5_out1 = self.conv2d5.forward(mul8_out1);
        let conv2d6_out1 = self.conv2d6.forward(conv2d5_out1.clone());
        let sigmoid3_out1 = burn::tensor::activation::sigmoid(conv2d6_out1.clone());
        let mul9_out1 = conv2d6_out1.mul(sigmoid3_out1);
        let conv2d7_out1 = self.conv2d7.forward(mul9_out1);
        let sigmoid4_out1 = burn::tensor::activation::sigmoid(conv2d7_out1.clone());
        let mul10_out1 = conv2d7_out1.mul(sigmoid4_out1);
        let conv2d8_out1 = self.conv2d8.forward(mul10_out1);
        let add3_out1 = conv2d8_out1.add(conv2d5_out1);
        let conv2d9_out1 = self.conv2d9.forward(add3_out1.clone());
        let sigmoid5_out1 = burn::tensor::activation::sigmoid(conv2d9_out1.clone());
        let mul11_out1 = conv2d9_out1.mul(sigmoid5_out1);
        let conv2d10_out1 = self.conv2d10.forward(mul11_out1);
        let sigmoid6_out1 = burn::tensor::activation::sigmoid(conv2d10_out1.clone());
        let mul12_out1 = conv2d10_out1.mul(sigmoid6_out1);
        let conv2d11_out1 = self.conv2d11.forward(mul12_out1);
        let add4_out1 = conv2d11_out1.add(add3_out1);
        let conv2d12_out1 = self.conv2d12.forward(add4_out1);
        let sigmoid7_out1 = burn::tensor::activation::sigmoid(conv2d12_out1.clone());
        let mul13_out1 = conv2d12_out1.mul(sigmoid7_out1);
        let conv2d13_out1 = self.conv2d13.forward(mul13_out1);
        let sigmoid8_out1 = burn::tensor::activation::sigmoid(conv2d13_out1.clone());
        let mul14_out1 = conv2d13_out1.mul(sigmoid8_out1);
        let transpose2_out1 = mul14_out1.clone().permute([0, 2, 3, 1]);
        let reducemean1_out1 = {
            transpose2_out1
                .mean_dim(1usize)
                .mean_dim(2usize)
                .squeeze_dims::<2usize>(&[1, 2])
        };
        let reshape5_out1 = reducemean1_out1.reshape([-1, 288, 1, 1]);
        let conv2d14_out1 = self.conv2d14.forward(reshape5_out1);
        let sigmoid9_out1 = burn::tensor::activation::sigmoid(conv2d14_out1.clone());
        let mul15_out1 = conv2d14_out1.mul(sigmoid9_out1);
        let reshape6_out1 = mul15_out1.reshape([-1, 18, 1, 1]);
        let conv2d15_out1 = self.conv2d15.forward(reshape6_out1);
        let sigmoid10_out1 = burn::tensor::activation::sigmoid(conv2d15_out1);
        let mul16_out1 = mul14_out1.mul(sigmoid10_out1);
        let conv2d16_out1 = self.conv2d16.forward(mul16_out1);
        let conv2d17_out1 = self.conv2d17.forward(conv2d16_out1.clone());
        let sigmoid11_out1 = burn::tensor::activation::sigmoid(conv2d17_out1.clone());
        let mul17_out1 = conv2d17_out1.mul(sigmoid11_out1);
        let conv2d18_out1 = self.conv2d18.forward(mul17_out1);
        let sigmoid12_out1 = burn::tensor::activation::sigmoid(conv2d18_out1.clone());
        let mul18_out1 = conv2d18_out1.mul(sigmoid12_out1);
        let transpose3_out1 = mul18_out1.clone().permute([0, 2, 3, 1]);
        let reducemean2_out1 = {
            transpose3_out1
                .mean_dim(1usize)
                .mean_dim(2usize)
                .squeeze_dims::<2usize>(&[1, 2])
        };
        let reshape7_out1 = reducemean2_out1.reshape([-1, 288, 1, 1]);
        let conv2d19_out1 = self.conv2d19.forward(reshape7_out1);
        let sigmoid13_out1 = burn::tensor::activation::sigmoid(conv2d19_out1.clone());
        let mul19_out1 = conv2d19_out1.mul(sigmoid13_out1);
        let reshape8_out1 = mul19_out1.reshape([-1, 18, 1, 1]);
        let conv2d20_out1 = self.conv2d20.forward(reshape8_out1);
        let sigmoid14_out1 = burn::tensor::activation::sigmoid(conv2d20_out1);
        let mul20_out1 = mul18_out1.mul(sigmoid14_out1);
        let conv2d21_out1 = self.conv2d21.forward(mul20_out1);
        let add5_out1 = conv2d21_out1.add(conv2d16_out1);
        let conv2d22_out1 = self.conv2d22.forward(add5_out1.clone());
        let sigmoid15_out1 = burn::tensor::activation::sigmoid(conv2d22_out1.clone());
        let mul21_out1 = conv2d22_out1.mul(sigmoid15_out1);
        let conv2d23_out1 = self.conv2d23.forward(mul21_out1);
        let sigmoid16_out1 = burn::tensor::activation::sigmoid(conv2d23_out1.clone());
        let mul22_out1 = conv2d23_out1.mul(sigmoid16_out1);
        let transpose4_out1 = mul22_out1.clone().permute([0, 2, 3, 1]);
        let reducemean3_out1 = {
            transpose4_out1
                .mean_dim(1usize)
                .mean_dim(2usize)
                .squeeze_dims::<2usize>(&[1, 2])
        };
        let reshape9_out1 = reducemean3_out1.reshape([-1, 288, 1, 1]);
        let conv2d24_out1 = self.conv2d24.forward(reshape9_out1);
        let sigmoid17_out1 = burn::tensor::activation::sigmoid(conv2d24_out1.clone());
        let mul23_out1 = conv2d24_out1.mul(sigmoid17_out1);
        let reshape10_out1 = mul23_out1.reshape([-1, 18, 1, 1]);
        let conv2d25_out1 = self.conv2d25.forward(reshape10_out1);
        let sigmoid18_out1 = burn::tensor::activation::sigmoid(conv2d25_out1);
        let mul24_out1 = mul22_out1.mul(sigmoid18_out1);
        let conv2d26_out1 = self.conv2d26.forward(mul24_out1);
        let add6_out1 = conv2d26_out1.add(add5_out1);
        let conv2d27_out1 = self.conv2d27.forward(add6_out1.clone());
        let sigmoid19_out1 = burn::tensor::activation::sigmoid(conv2d27_out1.clone());
        let mul25_out1 = conv2d27_out1.mul(sigmoid19_out1);
        let conv2d28_out1 = self.conv2d28.forward(mul25_out1);
        let sigmoid20_out1 = burn::tensor::activation::sigmoid(conv2d28_out1.clone());
        let mul26_out1 = conv2d28_out1.mul(sigmoid20_out1);
        let transpose5_out1 = mul26_out1.clone().permute([0, 2, 3, 1]);
        let reducemean4_out1 = {
            transpose5_out1
                .mean_dim(1usize)
                .mean_dim(2usize)
                .squeeze_dims::<2usize>(&[1, 2])
        };
        let reshape11_out1 = reducemean4_out1.reshape([-1, 288, 1, 1]);
        let conv2d29_out1 = self.conv2d29.forward(reshape11_out1);
        let sigmoid21_out1 = burn::tensor::activation::sigmoid(conv2d29_out1.clone());
        let mul27_out1 = conv2d29_out1.mul(sigmoid21_out1);
        let reshape12_out1 = mul27_out1.reshape([-1, 18, 1, 1]);
        let conv2d30_out1 = self.conv2d30.forward(reshape12_out1);
        let sigmoid22_out1 = burn::tensor::activation::sigmoid(conv2d30_out1);
        let mul28_out1 = mul26_out1.mul(sigmoid22_out1);
        let conv2d31_out1 = self.conv2d31.forward(mul28_out1);
        let add7_out1 = conv2d31_out1.add(add6_out1);
        let conv2d32_out1 = self.conv2d32.forward(add7_out1);
        let sigmoid23_out1 = burn::tensor::activation::sigmoid(conv2d32_out1.clone());
        let mul29_out1 = conv2d32_out1.mul(sigmoid23_out1);
        let conv2d33_out1 = self.conv2d33.forward(mul29_out1);
        let sigmoid24_out1 = burn::tensor::activation::sigmoid(conv2d33_out1.clone());
        let mul30_out1 = conv2d33_out1.mul(sigmoid24_out1);
        let transpose6_out1 = mul30_out1.clone().permute([0, 2, 3, 1]);
        let reducemean5_out1 = {
            transpose6_out1
                .mean_dim(1usize)
                .mean_dim(2usize)
                .squeeze_dims::<2usize>(&[1, 2])
        };
        let reshape13_out1 = reducemean5_out1.reshape([-1, 864, 1, 1]);
        let conv2d34_out1 = self.conv2d34.forward(reshape13_out1);
        let sigmoid25_out1 = burn::tensor::activation::sigmoid(conv2d34_out1.clone());
        let mul31_out1 = conv2d34_out1.mul(sigmoid25_out1);
        let reshape14_out1 = mul31_out1.reshape([-1, 27, 1, 1]);
        let conv2d35_out1 = self.conv2d35.forward(reshape14_out1);
        let sigmoid26_out1 = burn::tensor::activation::sigmoid(conv2d35_out1);
        let mul32_out1 = mul30_out1.mul(sigmoid26_out1);
        let conv2d36_out1 = self.conv2d36.forward(mul32_out1);
        let conv2d37_out1 = self.conv2d37.forward(conv2d36_out1.clone());
        let sigmoid27_out1 = burn::tensor::activation::sigmoid(conv2d37_out1.clone());
        let mul33_out1 = conv2d37_out1.mul(sigmoid27_out1);
        let conv2d38_out1 = self.conv2d38.forward(mul33_out1);
        let sigmoid28_out1 = burn::tensor::activation::sigmoid(conv2d38_out1.clone());
        let mul34_out1 = conv2d38_out1.mul(sigmoid28_out1);
        let transpose7_out1 = mul34_out1.clone().permute([0, 2, 3, 1]);
        let reducemean6_out1 = {
            transpose7_out1
                .mean_dim(1usize)
                .mean_dim(2usize)
                .squeeze_dims::<2usize>(&[1, 2])
        };
        let reshape15_out1 = reducemean6_out1.reshape([-1, 864, 1, 1]);
        let conv2d39_out1 = self.conv2d39.forward(reshape15_out1);
        let sigmoid29_out1 = burn::tensor::activation::sigmoid(conv2d39_out1.clone());
        let mul35_out1 = conv2d39_out1.mul(sigmoid29_out1);
        let reshape16_out1 = mul35_out1.reshape([-1, 27, 1, 1]);
        let conv2d40_out1 = self.conv2d40.forward(reshape16_out1);
        let sigmoid30_out1 = burn::tensor::activation::sigmoid(conv2d40_out1);
        let mul36_out1 = mul34_out1.mul(sigmoid30_out1);
        let conv2d41_out1 = self.conv2d41.forward(mul36_out1);
        let add8_out1 = conv2d41_out1.add(conv2d36_out1);
        let conv2d42_out1 = self.conv2d42.forward(add8_out1.clone());
        let sigmoid31_out1 = burn::tensor::activation::sigmoid(conv2d42_out1.clone());
        let mul37_out1 = conv2d42_out1.mul(sigmoid31_out1);
        let conv2d43_out1 = self.conv2d43.forward(mul37_out1);
        let sigmoid32_out1 = burn::tensor::activation::sigmoid(conv2d43_out1.clone());
        let mul38_out1 = conv2d43_out1.mul(sigmoid32_out1);
        let transpose8_out1 = mul38_out1.clone().permute([0, 2, 3, 1]);
        let reducemean7_out1 = {
            transpose8_out1
                .mean_dim(1usize)
                .mean_dim(2usize)
                .squeeze_dims::<2usize>(&[1, 2])
        };
        let reshape17_out1 = reducemean7_out1.reshape([-1, 864, 1, 1]);
        let conv2d44_out1 = self.conv2d44.forward(reshape17_out1);
        let sigmoid33_out1 = burn::tensor::activation::sigmoid(conv2d44_out1.clone());
        let mul39_out1 = conv2d44_out1.mul(sigmoid33_out1);
        let reshape18_out1 = mul39_out1.reshape([-1, 27, 1, 1]);
        let conv2d45_out1 = self.conv2d45.forward(reshape18_out1);
        let sigmoid34_out1 = burn::tensor::activation::sigmoid(conv2d45_out1);
        let mul40_out1 = mul38_out1.mul(sigmoid34_out1);
        let conv2d46_out1 = self.conv2d46.forward(mul40_out1);
        let add9_out1 = conv2d46_out1.add(add8_out1);
        let conv2d47_out1 = self.conv2d47.forward(add9_out1.clone());
        let sigmoid35_out1 = burn::tensor::activation::sigmoid(conv2d47_out1.clone());
        let mul41_out1 = conv2d47_out1.mul(sigmoid35_out1);
        let conv2d48_out1 = self.conv2d48.forward(mul41_out1);
        let sigmoid36_out1 = burn::tensor::activation::sigmoid(conv2d48_out1.clone());
        let mul42_out1 = conv2d48_out1.mul(sigmoid36_out1);
        let transpose9_out1 = mul42_out1.clone().permute([0, 2, 3, 1]);
        let reducemean8_out1 = {
            transpose9_out1
                .mean_dim(1usize)
                .mean_dim(2usize)
                .squeeze_dims::<2usize>(&[1, 2])
        };
        let reshape19_out1 = reducemean8_out1.reshape([-1, 864, 1, 1]);
        let conv2d49_out1 = self.conv2d49.forward(reshape19_out1);
        let sigmoid37_out1 = burn::tensor::activation::sigmoid(conv2d49_out1.clone());
        let mul43_out1 = conv2d49_out1.mul(sigmoid37_out1);
        let reshape20_out1 = mul43_out1.reshape([-1, 27, 1, 1]);
        let conv2d50_out1 = self.conv2d50.forward(reshape20_out1);
        let sigmoid38_out1 = burn::tensor::activation::sigmoid(conv2d50_out1);
        let mul44_out1 = mul42_out1.mul(sigmoid38_out1);
        let conv2d51_out1 = self.conv2d51.forward(mul44_out1);
        let add10_out1 = conv2d51_out1.add(add9_out1);
        let conv2d52_out1 = self.conv2d52.forward(add10_out1.clone());
        let sigmoid39_out1 = burn::tensor::activation::sigmoid(conv2d52_out1.clone());
        let mul45_out1 = conv2d52_out1.mul(sigmoid39_out1);
        let conv2d53_out1 = self.conv2d53.forward(mul45_out1);
        let sigmoid40_out1 = burn::tensor::activation::sigmoid(conv2d53_out1.clone());
        let mul46_out1 = conv2d53_out1.mul(sigmoid40_out1);
        let transpose10_out1 = mul46_out1.clone().permute([0, 2, 3, 1]);
        let reducemean9_out1 = {
            transpose10_out1
                .mean_dim(1usize)
                .mean_dim(2usize)
                .squeeze_dims::<2usize>(&[1, 2])
        };
        let reshape21_out1 = reducemean9_out1.reshape([-1, 864, 1, 1]);
        let conv2d54_out1 = self.conv2d54.forward(reshape21_out1);
        let sigmoid41_out1 = burn::tensor::activation::sigmoid(conv2d54_out1.clone());
        let mul47_out1 = conv2d54_out1.mul(sigmoid41_out1);
        let reshape22_out1 = mul47_out1.reshape([-1, 27, 1, 1]);
        let conv2d55_out1 = self.conv2d55.forward(reshape22_out1);
        let sigmoid42_out1 = burn::tensor::activation::sigmoid(conv2d55_out1);
        let mul48_out1 = mul46_out1.mul(sigmoid42_out1);
        let conv2d56_out1 = self.conv2d56.forward(mul48_out1);
        let add11_out1 = conv2d56_out1.add(add10_out1);
        let conv2d57_out1 = self.conv2d57.forward(add11_out1);
        let sigmoid43_out1 = burn::tensor::activation::sigmoid(conv2d57_out1.clone());
        let mul49_out1 = conv2d57_out1.mul(sigmoid43_out1);
        let conv2d58_out1 = self.conv2d58.forward(mul49_out1);
        let sigmoid44_out1 = burn::tensor::activation::sigmoid(conv2d58_out1.clone());
        let mul50_out1 = conv2d58_out1.mul(sigmoid44_out1);
        let transpose11_out1 = mul50_out1.clone().permute([0, 2, 3, 1]);
        let reducemean10_out1 = {
            transpose11_out1
                .mean_dim(1usize)
                .mean_dim(2usize)
                .squeeze_dims::<2usize>(&[1, 2])
        };
        let reshape23_out1 = reducemean10_out1.reshape([-1, 1536, 1, 1]);
        let conv2d59_out1 = self.conv2d59.forward(reshape23_out1);
        let sigmoid45_out1 = burn::tensor::activation::sigmoid(conv2d59_out1.clone());
        let mul51_out1 = conv2d59_out1.mul(sigmoid45_out1);
        let reshape24_out1 = mul51_out1.reshape([-1, 48, 1, 1]);
        let conv2d60_out1 = self.conv2d60.forward(reshape24_out1);
        let sigmoid46_out1 = burn::tensor::activation::sigmoid(conv2d60_out1);
        let mul52_out1 = mul50_out1.mul(sigmoid46_out1);
        let conv2d61_out1 = self.conv2d61.forward(mul52_out1);
        let conv2d62_out1 = self.conv2d62.forward(conv2d61_out1.clone());
        let sigmoid47_out1 = burn::tensor::activation::sigmoid(conv2d62_out1.clone());
        let mul53_out1 = conv2d62_out1.mul(sigmoid47_out1);
        let conv2d63_out1 = self.conv2d63.forward(mul53_out1);
        let sigmoid48_out1 = burn::tensor::activation::sigmoid(conv2d63_out1.clone());
        let mul54_out1 = conv2d63_out1.mul(sigmoid48_out1);
        let transpose12_out1 = mul54_out1.clone().permute([0, 2, 3, 1]);
        let reducemean11_out1 = {
            transpose12_out1
                .mean_dim(1usize)
                .mean_dim(2usize)
                .squeeze_dims::<2usize>(&[1, 2])
        };
        let reshape25_out1 = reducemean11_out1.reshape([-1, 1536, 1, 1]);
        let conv2d64_out1 = self.conv2d64.forward(reshape25_out1);
        let sigmoid49_out1 = burn::tensor::activation::sigmoid(conv2d64_out1.clone());
        let mul55_out1 = conv2d64_out1.mul(sigmoid49_out1);
        let reshape26_out1 = mul55_out1.reshape([-1, 48, 1, 1]);
        let conv2d65_out1 = self.conv2d65.forward(reshape26_out1);
        let sigmoid50_out1 = burn::tensor::activation::sigmoid(conv2d65_out1);
        let mul56_out1 = mul54_out1.mul(sigmoid50_out1);
        let conv2d66_out1 = self.conv2d66.forward(mul56_out1);
        let add12_out1 = conv2d66_out1.add(conv2d61_out1);
        let conv2d67_out1 = self.conv2d67.forward(add12_out1.clone());
        let sigmoid51_out1 = burn::tensor::activation::sigmoid(conv2d67_out1.clone());
        let mul57_out1 = conv2d67_out1.mul(sigmoid51_out1);
        let conv2d68_out1 = self.conv2d68.forward(mul57_out1);
        let sigmoid52_out1 = burn::tensor::activation::sigmoid(conv2d68_out1.clone());
        let mul58_out1 = conv2d68_out1.mul(sigmoid52_out1);
        let transpose13_out1 = mul58_out1.clone().permute([0, 2, 3, 1]);
        let reducemean12_out1 = {
            transpose13_out1
                .mean_dim(1usize)
                .mean_dim(2usize)
                .squeeze_dims::<2usize>(&[1, 2])
        };
        let reshape27_out1 = reducemean12_out1.reshape([-1, 1536, 1, 1]);
        let conv2d69_out1 = self.conv2d69.forward(reshape27_out1);
        let sigmoid53_out1 = burn::tensor::activation::sigmoid(conv2d69_out1.clone());
        let mul59_out1 = conv2d69_out1.mul(sigmoid53_out1);
        let reshape28_out1 = mul59_out1.reshape([-1, 48, 1, 1]);
        let conv2d70_out1 = self.conv2d70.forward(reshape28_out1);
        let sigmoid54_out1 = burn::tensor::activation::sigmoid(conv2d70_out1);
        let mul60_out1 = mul58_out1.mul(sigmoid54_out1);
        let conv2d71_out1 = self.conv2d71.forward(mul60_out1);
        let add13_out1 = conv2d71_out1.add(add12_out1);
        let conv2d72_out1 = self.conv2d72.forward(add13_out1.clone());
        let sigmoid55_out1 = burn::tensor::activation::sigmoid(conv2d72_out1.clone());
        let mul61_out1 = conv2d72_out1.mul(sigmoid55_out1);
        let conv2d73_out1 = self.conv2d73.forward(mul61_out1);
        let sigmoid56_out1 = burn::tensor::activation::sigmoid(conv2d73_out1.clone());
        let mul62_out1 = conv2d73_out1.mul(sigmoid56_out1);
        let transpose14_out1 = mul62_out1.clone().permute([0, 2, 3, 1]);
        let reducemean13_out1 = {
            transpose14_out1
                .mean_dim(1usize)
                .mean_dim(2usize)
                .squeeze_dims::<2usize>(&[1, 2])
        };
        let reshape29_out1 = reducemean13_out1.reshape([-1, 1536, 1, 1]);
        let conv2d74_out1 = self.conv2d74.forward(reshape29_out1);
        let sigmoid57_out1 = burn::tensor::activation::sigmoid(conv2d74_out1.clone());
        let mul63_out1 = conv2d74_out1.mul(sigmoid57_out1);
        let reshape30_out1 = mul63_out1.reshape([-1, 48, 1, 1]);
        let conv2d75_out1 = self.conv2d75.forward(reshape30_out1);
        let sigmoid58_out1 = burn::tensor::activation::sigmoid(conv2d75_out1);
        let mul64_out1 = mul62_out1.mul(sigmoid58_out1);
        let conv2d76_out1 = self.conv2d76.forward(mul64_out1);
        let add14_out1 = conv2d76_out1.add(add13_out1);
        let mul65_out1 = add14_out1.mul(constant140_out1);
        let add15_out1 = mul65_out1.add(constant141_out1);
        let relu2_out1 = burn::tensor::activation::relu(add15_out1);
        let conv2d77_out1 = self.conv2d77.forward(relu2_out1);
        let relu3_out1 = burn::tensor::activation::relu(conv2d77_out1);
        let transpose15_out1 = relu3_out1.permute([0, 2, 3, 1]);
        let reducemean14_out1 = {
            transpose15_out1
                .mean_dim(1usize)
                .mean_dim(2usize)
                .squeeze_dims::<2usize>(&[1, 2])
        };
        let linear5_out1 = self.linear5.forward(reducemean14_out1);
        linear5_out1
    }
}