https://nl2code.github.io/posts/TorchDataEval/