Run problem in face age estimation

Part 1 (2019) Advanced (Part 1 v3)
1

when i want to make a age estimation with the dataset from UTKFace , and i use the linear regression result as a out layer, and when i run i have the error:

RuntimeError: Expected object of scalar type Float but got scalar type Long for argument #2 ‘target’

and my code is like the picture:

5
5.png1920×1033 50.9 KB

6
6.png1920×1033 48.4 KB

7
7.png1920×1033 56.5 KB

does any one know how to fix it , thank you. my email is 2242988415@qq.com.

2

Your targets should be type float. So convert them to torch.float in the dataset.