mlr3torch
mlr3torch copied to clipboard
Invalid shape: better error handling
Better error message: Say which shape is expected.
Also, the predefiend networks should add to their documentation which input shapes they expect for the lazy tensor inputs.
Also add checks on input type (float32 in most cases)