search

rwth-i6 / pytorch-to-returnn

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

fix for axes in GenericPadNd #118

Closed vieting closed 2 years ago

vieting commented 2 years ago

In GenericPadNd, the axes are currently converted to RETURNN manually before passing them to self._get_input_axis_to_returnn, however, self._get_input_axis_to_returnn should receive the torch axes. This is fixed here, which also simplifies the code a bit.

vieting commented 2 years ago

We first need to fix #117 and then rebase.