This repository was archived by the owner on Jul 14, 2020. It is now read-only.
Add a way to customize get_batch_class #10
Description
For instance for graph, get_batch_class would need to add another attr for remembering batch index.