动态图定义模型时如何获得batch size?
收藏
在用谱归一化或者写nonlocal模块时,需要在模型的构造函数里定义子层时输入batch_size或其他跟数据形状有关的参数。请问如何在模型定义类的构造函数里获得特征图的形状参数?
0
收藏
请登录后评论
得到特征图的形状在 forward 方法里很简单,如何在构造函数里就的到呢?
看过一个项目,在1.8版本里是可以得到的,在2.0版本里怎么写呢?
请教各位大佬们
还是说这个做法比较“违和”,不提倡呢?
现在动态图静态图一键互转是不是只能在模型导出的时候?
我理解是,就是读写参数时转换
我还发现一问题。gradcam前向计算时得用eval模式,不然bn层会影响预测结果。但是还得取梯度,这就不好办了。。。
eval模式下不能计算梯度,但train模式下前向计算的结果又不对。。。
不行,看看官方的可视化套件怎么写的吧
对了,说不定那也需要取特征图的形状~