shouldn't it be F.cross_entropy instead?
shouldn't it be F.cross_entropy instead?