首页 炼丹房 帖子详情
动态图定义模型时如何获得batch size?
收藏
快速回复
炼丹房 问答新手上路 498 10
动态图定义模型时如何获得batch size?
收藏
快速回复
炼丹房 问答新手上路 498 10

在用谱归一化或者写nonlocal模块时,需要在模型的构造函数里定义子层时输入batch_size或其他跟数据形状有关的参数。请问如何在模型定义类的构造函数里获得特征图的形状参数?

0
收藏
回复
全部评论(10)
时间顺序
AIStudio810258
#2 回复于2021-03

得到特征图的形状在 forward 方法里很简单,如何在构造函数里就的到呢?

0
回复
AIStudio810258
#3 回复于2021-03

看过一个项目,在1.8版本里是可以得到的,在2.0版本里怎么写呢?

0
回复
AIStudio810258
#4 回复于2021-03

请教各位大佬们

0
回复
AIStudio810258
#5 回复于2021-03

还是说这个做法比较“违和”,不提倡呢?

0
回复
AIStudio810260
#6 回复于2021-03

现在动态图静态图一键互转是不是只能在模型导出的时候?

0
回复
AIStudio810258
#7 回复于2021-04
现在动态图静态图一键互转是不是只能在模型导出的时候?

我理解是,就是读写参数时转换

0
回复
AIStudio810258
#8 回复于2021-04
现在动态图静态图一键互转是不是只能在模型导出的时候?

我还发现一问题。gradcam前向计算时得用eval模式,不然bn层会影响预测结果。但是还得取梯度,这就不好办了。。。

0
回复
AIStudio810258
#9 回复于2021-04

eval模式下不能计算梯度,但train模式下前向计算的结果又不对。。。

0
回复
AIStudio810258
#10 回复于2021-04

不行,看看官方的可视化套件怎么写的吧

0
回复
AIStudio810258
#11 回复于2021-04

对了,说不定那也需要取特征图的形状~

0
回复
在@后输入用户全名并按空格结束,可艾特全站任一用户