CELoss may be wrong?

[Feng1909]

Hello:
I find some code that I think may be wrong
return nn.CrossEntropyLoss()(x, target_age)
https://github.com/InterDigitalInc/HRFAE/blob/master/trainer.py#L62

which should be
return nn.CrossEntropyLoss(x, target_age)
?

I don't know if it has some impact.

[Feng1909]

my bad