search

rwth-i6 / pytorch-to-returnn

Make PyTorch code runnable within RETURNN
3 stars 6 forks source link

Additions for compatibility with newest fairseq version #122

Closed LabChameleon closed 2 years ago

LabChameleon commented 2 years ago

I made small additions to achieve compatibility with the current fairseq version for wav2vec2 (commit: 31d94f556bd49bc7e61511adbda482b2c54652b5).

torch.jit.is_tracing I set to False similar to the existing implementation of torch.jit.is_scripting. I think this is fine because torch.jit is never really relevant for the usecase of pytorch_to_returnn, or is it? . What do you think?

@vieting @albertz