mlr3torch
mlr3torch copied to clipboard
`nn_ft_transformer_block` should just be a single block, not multiple blocks. Seperate the output head in its own `nn_module`