Closed vieting closed 1 year ago
I added test_dct
as an example test case because that's what I was interested in. We can also simplify in general or reformat the code in dct()
if you don't like it.
There are three failing tests. Did you check them?
There are three failing tests. Did you check them?
Yes, they are the same as in #125.
I'd like to create a module which has a torch forward function, but nevertheless is mapped to a custom RETURNN layer call. The basic idea is to have something like
Right now, the custom
create_returnn_layer_dict
is ignored and the code insideforward
is wrapped. This PR changes this behavior to make sure to use the customcreate_returnn_layer_dict
.