首页 Paddle框架 帖子详情
怎样在program中实时获取bn的mean和variance值
收藏
快速回复
Paddle框架 问答深度学习模型训练炼丹技巧 763 2
怎样在program中实时获取bn的mean和variance值
收藏
快速回复
Paddle框架 问答深度学习模型训练炼丹技巧 763 2

如下所示,我想在bn之后获取当前的moving_mean和moving_variance的值(不是基于fetch list),再基于该值做一些额外的操作,想问下这个怎么实现呢?

temp_y = fluid.layers.batch_norm(
            input=conv,
            act=act,
            param_attr=ParamAttr(name=bn_name + '_scale'),
            bias_attr=ParamAttr(bn_name + '_offset'),
            moving_mean_name=bn_name + '_mean',
            moving_variance_name=bn_name + '_variance')
y = some_func(moving_mean, moving_variance, temp_y)
0
收藏
回复
全部评论(2)
时间顺序
AIStudio792089
#2 回复于2020-01

可以试下根据moving_mean_name和moving_variance_name,使用fluid.get_default_main_program().blocks[0].var("var_name")获取到这两个variable,然后就可以像其他内容一样使用了。

0
回复
AIStudio374196
#3 回复于2020-01

嗯嗯试了下,应该是fluid.default_main_program().blocks[0].var("var_name"),多谢啦

0
回复
需求/bug反馈?一键提issue告诉我们
发现bug?如果您知道修复办法,欢迎提pr直接参与建设飞桨~
在@后输入用户全名并按空格结束,可艾特全站任一用户