RenMin1991/Dyamic-Graph-Representation

关于深层网络的迁移问题

zzh-ecnu opened this issue · 1 comments

你好~

我看到在README里有人脸识别的结果,包括resnet50和resnet101,但是code里好像是一个5层CNN。

所以想请教一下对于resnet的深层网络,怎么将DGR迁移过去,需要做出哪些改变?

谢谢~

需要在以下三个方面进行出调整:

  1. 调整Graph模块的插入位置,对于resnet而言,可以在最后一个(single-scale)或三个(multi-scale)block之后插入Graph模块;
  2. 调整Graph的节点数量,其所插入的feature map尺寸越大,所需要的节点数量越多,例如当前feature map的尺寸为 H X W,可以将节点数量设置为 2*(H*W)^0.5附近的数值。
  3. 调整Graph模块的初始化参数,我们在训练中发现模型性能会在一定程度上受到初始化参数的影响,具体可以在 model/model_singlescale.py 中的 GraphGen 类的 _init_params 函数中进行设置。
    希望有所帮助!