Hi,
Is there a way which I can apply to() function to a Dataset then to() will be applied to the data in Dataset ?
It is a lot more convenient if  I can make a nn.Module to() some device meanwhile the data nn.Module used.