lhwcv / DTLN_pytorch

Dual-signal Transformation LSTM Network, PyTorch,NCNN
Apache License 2.0
69 stars 15 forks source link

Is model.pth converted from the tensorflow pre-trained model or trained by yourself? #3

Closed zuowanbushiwo closed 2 years ago

zuowanbushiwo commented 2 years ago

Dear sir Because there is no training related code in this project, So how did this model come about? Can DTLN-aec this be ported to ncnn? Thanks!

zuowanbushiwo commented 2 years ago

@lhwcv
Can add some script about pytorch model or onnx model convert to ncnn model ? Thanks!

lhwcv commented 2 years ago

Thank you for your issue.

zuowanbushiwo commented 2 years ago

Dear sir Thanks,you are so friendly .

  • I converted from TensorFlow pertained one, we can also retrain with PyTorch.

Can you also share some scripts to convert TensorFlow pertained to PyTorch model.pth file? which tensorFlow pertained are you using, there are xx.h5 , xx.pb and xx.tflte three different types of pertained files 。

and about DTLN-aec , The author only opened the pre-trained model of tflite, can this also be converted into a pytorch mode directly? then convert to onnx and ncnn. Or need us to retrain with pytorch first?

Looking forward to your reply

lhwcv commented 2 years ago

@zuowanbushiwo Hi,

zuowanbushiwo commented 2 years ago

@lhwcv

  • I'll share my script converted from xx.h5 file this week.

thanks, watching!

-tflite model can also be converted to PyTorch model, but I not tried. If you have trained one, it is more convenient..

I'm a beginer in deep learning and the training environment has not yet been set up . I found one here yesterday, but the effect is not as good as the original author's.

lhwcv commented 2 years ago

@zuowanbushiwo

zuowanbushiwo commented 2 years ago

@lhwcv 👍