PaddlePaddle如何只进行卷积运算不训练
小潴潴343 发布于2019-07 浏览:1617 回复:2
0
收藏

想问一下PaddlePaddle如何只进行卷积运算不训练,我想了一个算子,我要用它处理几万张图片,分辨率很高,现在用CPU跑要5秒一张。想用这个免费的GPU加速一下,所以怎样才能不训练,只进行卷积运算能用上这张免费的GPU,(每日签到的时长我己经浪费上百个小时了o(╯□╰)o)

收藏
点赞
0
个赞
共2条回复 最后由星光ld1回复于2020-01
#3星光ld1回复于2020-01

对于那一层把训练梯度设置`stop_gradient=True`就可以吧?

参考(https://www.paddlepaddle.org.cn/documentation/docs/zh/faq/train_cn.html#q-stop-gradient-true)

如果是定义的模型应该只要把mode改为eval模式就可以

0
#2625076229回复于2019-12

好像可以把variable调成不训练

0
TOP
切换版块