returnn.frontend.conversions.torch_nn

Import some of the torch.nn modules.

returnn.frontend.conversions.torch_nn.import_params_torch_linear_to_rf(model_pt: torch.nn.Linear, model_rf: rf.Linear)[source]

import params from torch.nn.Linear to rf.Linear

returnn.frontend.conversions.torch_nn.import_params_torch_conv1d_to_rf(model_pt: torch.nn.Conv1d, model_rf: rf.Conv1d)[source]

import params from torch.nn.Conv1d to rf.Conv1d

returnn.frontend.conversions.torch_nn.import_params_torch_layer_norm_to_rf(model_pt: torch.nn.LayerNorm, model_rf: rf.LayerNorm)[source]

Import the parameters from torch.nn.LayerNorm to rf.LayerNorm.