使用PaddleSlim进行uniform剪裁mobilenet v2模型,训练loss出现nan

798qvoo8  于 2021-11-29  发布在  Java
关注(0)|答案(2)|浏览(545)

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

剪裁配置如下:

日志:

t5fffqht

t5fffqht1#

你好,尝试过常规的处理nan的方法吗,比如减小learnin_rate之类的。

另外,issue能,能贴文字的,建议尽量别贴图哈~

相关问题