Open urig opened 1 year ago
Thanks for opening the issue.
Have you looked further into this ? Digged the unit tests ? versions of Pytorch ? etc ?
Hi @mpariente. Thanks for responding. The version of PyTorch is 1.9.1. I'll try and clear some time to go over the tests and maybe create a failing test for this. Cheers, Uri
🐛 Bug
When trying to convert a pre-trained instance of
BaseEncoderMaskerDecoder
to TorchScript aRuntimeError
is thrown, seemingly because of incompatible code insideasteroid
'sConv1DBlocks
class.To Reproduce
Run this bit of code where
asteroid
v0.6.0 is installed:Expected behavior
model_scripted
should contain the TorchScript equivalent of the model.Actual behavior
A
RuntimeError
is raised from the last line of code:Examining https://github.com/asteroid-team/asteroid/blob/v0.6.0/asteroid/masknn/convolutional.py#L90 indeed shows two
return
statements each returning a different type.Environment
Package versions
Additional info
Reproduced on Windows with Python 3.9.6.