目标:只保留前向功能,尽可能减少代码,以及内存、显存占用。

步骤:

1、  只保留常用的类,包括cpp、cu、hpp文件。

2、  删除类中以Backward 为前缀的函数。

3、  如果使用了cudnn,把CUDNN_STREAMS_PER_GROUP 设置成1。

4、  由于BatchNorm等层的中间结果是独立的,即一个模型中如果有多个BatchNorm层,会产生多份中间变量,可以修改Forward 函数,增加一个保存中间结果的参数,达到复用效果。

5、  去除net类中的大部分成员变量,以及相关的函数。

简化Caffe

相关文章:

  • 2021-11-27
  • 2021-07-07
  • 2022-12-23
  • 2021-09-04
  • 2021-07-08
  • 2021-07-18
  • 2021-05-20
  • 2021-06-13
猜你喜欢
  • 2021-12-12
  • 2021-06-24
  • 2021-08-25
  • 2022-01-05
  • 2021-11-18
  • 2022-12-23
相关资源
相似解决方案