I got a training result as below:
targets: tensor([[ 80.],
[ 71.],
[ 65.],
[149.]], device='cuda:0'),
pred: tensor([[33.4143],
[ nan],
[32.8215],
[32.5122]], device='cuda:0', grad_fn=)
I followed your introduction to do all the preprocess and trained the model with JHU dataset. Is it a random error?
I got a training result as below:
targets: tensor([[ 80.],
[ 71.],
[ 65.],
[149.]], device='cuda:0'),
pred: tensor([[33.4143],
[ nan],
[32.8215],
[32.5122]], device='cuda:0', grad_fn=)
I followed your introduction to do all the preprocess and trained the model with JHU dataset. Is it a random error?