使用PaddleSlim进行uniform剪裁mobilenet v2模型,训练loss出现nan
Created by: kk12333
1)PaddlePaddle版本:paddle-fluid-v1.4.1
2)CPU:无
3)GPU:cuda8.0,cudnn7.5
4)系统环境:CentOS release 6.10
- 训练信息
1)单机4卡 P40
2)显存:单P40卡显存16G左右
3)op情况:官网mobilenet v2结构涉及的op
- 问题描述:用PaddleSlim官网uniform pruning的demo,修改为mobilenet v2模型,并修改对应的.yaml配置,直接启动剪裁,finetune出现loss nan
剪裁配置如下:

日志:
