Pytorch Check If Model Is In Eval Mode