float32 & float64 Error

[24kobebryant]

I just run this code, and it occurs the problem that

found the dtype double but expect float

and I found this code may have some problems:

target = batch["val_acc"]
predict = net(batch)
loss = criterion(predict, target)

The dtype of target is float64 but the predict is float32.

Have I made some mistakes?

[yAya-yns]

I got the same error when running the python train.py

[hfjchdjdh]

me too,anyone knows?