Rasa test nlu batch size

Hello, I am struggling to configure the DIET classifier for test process… My nlu.yml file contains around 100k lines. After splitting, there are 90k for train and 10k for test. When I train the model with the default batch size (60-256), no problem, it takes 2h with my machine. But when I am testing, which is supposed to go much faster, it takes ages because it evaluates the data example per example… How to change the behaviour so the rasa test also does batching ?