簡化Caffe

目標:只保留前向功能,儘可能減少代碼,以及內存、顯存佔用。 步驟: 1、  只保留常用的類,包括cpp、cu、hpp文件。 2、  刪除類中以Backward 爲前綴的函數。 3、  如果使用了cudnn,把CUDNN_STREAMS_PER_GROUP 設置成1。 4、  由於BatchNorm等層的中間結果是獨立的,即一個模型中如果有多個BatchNorm層,會產生多份中間變量,可以修改For
相關文章
相關標籤/搜索