I am currently trying train a CNN using PyTorch to predict a subject\'s age. The age group ranges from 0 to 116. I used the same model to train it on gender classification w