mlr3torch
mlr3torch copied to clipboard
Deep learning framework for the mlr3 ecosystem based on torch
TODOs
* How to use predefined architectures * How slow is it?
the nn_linear_0 in the readme should not be necessary and is not transparent
Implement: https://torch.mlverse.org/docs/reference/torch_adaptive_avg_pool1d.html?q=adaptive#null for 1d, 2d and 3d. This is similar to: https://github.com/mlr-org/mlr3torch/blob/main/R/PipeOpTorchAvgPool.R
https://github.com/mlr-org/mlr-org.r-universe.dev/blob/main/packages.json
* GitHub has GPU support on [their roadmap](https://github.com/github/roadmap/issues/505) but it is not clear when this will be available.