mlr3torch
mlr3torch copied to clipboard
Offer a way to implement a `PipeOpTorch` without implementing a module
When we implemented the FTTransformer, it feels like a lot of unnecessary/duplicated code had to be written.
If one wants to work with mlr3torch, we should have a convenience function that makes the creation of custom PipeOpTorch objects from non-existing nn_modules easier.