search

RookieJunChen / Inter-SubNet

The official PyTorch implementation of "Inter-SubNet: Speech Enhancement with Subband Interaction", accepted by ICASSP 2023.
Apache License 2.0
91 stars 12 forks source link

convert onnnx model #7

Open lightwithshadow opened 7 months ago

lightwithshadow commented 7 months ago

Thank you for your excellent work， How to export Pytorch model in ONNX format？thx

RookieJunChen commented 7 months ago

Sorry, I haven't tried the conversion to onnx format for the Inter-SubNet. You might try to make an attempt in the speech_enhance/inter_subnet/model/Inter_SubNet.py code file, using the onnx tool library that comes with torch?