`ThunderModule` does not convert targets to the same type as outputs of the model.

[arseniybelkov]

In inference_step, y remains the same type after being received from data loader. However x is guaranteed to be np.ndarray after being processed by architecture.