pytorch cross entropy training trick

NoSuchKey