训练测试

[CompliceU]

@xialuxi 你好，非常感谢你的工作，我最近也在模仿你的代码进行人脸关键点的开发，训练代码是能够正常运行的，但是每代训练后的测试会报错，如下：

显示的是维度不匹配，不知道这个需要怎么解决

[xialuxi]

测试的代码，得到模型输出的时候， 把关键点部分截掉。只验证box就行。