How to write a call back to convert datatype?

[jS5t3r]

How to change the data type of the labels?

This code snippet:

for data, label in dataloader:
    data = data.cuda()
    data = labels.float32().cuda() # here I change to float32

should be called when execute .fit()