pytorch测试loss的简易方法