[FEATURE] Try to make all fit and predict methods take the same type of input
hlums opened this issue · comments
Hong Lu commented
Description
Decide if all fit and predict methods should take dataset or dataloaders and update the methods as needed.