Hi,
I did take a look but unfortunately I am not familiar with torchtext and the custom dataloader used there.
Similarly the Batch object is custom from that repo.
I guess you should check the repo’s source code to see how to work properly with this Batch object (how to extract data from it / move it to a different device).