PaddlePaddle如何只进行卷积运算不训练
0
收藏
想问一下PaddlePaddle如何只进行卷积运算不训练，我想了一个算子，我要用它处理几万张图片，分辨率很高，现在用CPU跑要5秒一张。想用这个免费的GPU加速一下，所以怎样才能不训练，只进行卷积运算能用上这张免费的GPU，（每日签到的时长我己经浪费上百个小时了o(╯□╰)o）
收藏
点赞
0
个赞
请登录后评论
TOP
切换版块
对于那一层把训练梯度设置`stop_gradient=True`就可以吧？
参考(https://www.paddlepaddle.org.cn/documentation/docs/zh/faq/train_cn.html#q-stop-gradient-true)
如果是定义的模型应该只要把mode改为eval模式就可以
好像可以把variable调成不训练