Fix train_utils_test bug for not comprehensive tests #45
Description
Currently, config.data.kwargs is not tested.
In addition, would be very useful to have data.kwargs automatically loading from dataset rather than being specified by hand
{{ message }}