Following up on this as I implemented the .to() in the forward function. If I hope to use DataParallel, does this method not work anymore? If so, why is that the case? Do I need to figure out a way to construct my data such that I send it to the device before being fed into the forward function?
Thanks!