search

mlr-org / mlr3torch

Deep learning framework for the mlr3 ecosystem based on torch
https://mlr3torch.mlr-org.com
Other
40 stars 7 forks source link

add small autotest #279

Closed sebffischer closed 2 months ago

sebffischer commented 2 months ago

just check that learner can be trained and predicted, very small batch size. Just check that dimensions fit.