Open cyrusvahidi opened 8 months ago
torch.jit.script throws a cannot create weak reference to 'numpy.ufunc' object error when compiled LogmelFilterBank. I have fixed this by changing numpy operations to torch (np.maximum) and storing self.ref and self.amin as tensors.
torch.jit.script
cannot create weak reference to 'numpy.ufunc' object
LogmelFilterBank
self.ref
self.amin
torch.jit.script
throws acannot create weak reference to 'numpy.ufunc' object
error when compiledLogmelFilterBank
. I have fixed this by changing numpy operations to torch (np.maximum) and storingself.ref
andself.amin
as tensors.