Thanks for the information.
Could you add a print statement to BBoxTransform and check the device of these tensors?
I would assume that self.mean and self.std are not pushed to the device correctly (line of code 1), as they are not registered as a buffer or parameter.