def _unpack_minibatch(self):
        """HuggingFace minibatches are dictionaries of tensors.
        Move tensors to the current device."""
        for k in self.mbatch.keys():
            self.mbatch[k] = self.mbatch[k].to(self.device)
HuggingFace minibatches are dictionaries of tensors.
        Move tensors to the current device.
examples/nlp.py