Memory-efficient Implementation of DenseNets

Memory-efficient Implementation of DenseNets 改進densenet模型佔用顯存較大的問題。 對於特徵的極致利用可以提高模型的表現能力,用時由於生成大量的intermediate feature(中間特徵),因此存儲這些intermediate feature會佔用大量的顯存。爲了能夠在GPU下跑更深的densenet網絡,這篇文章通過對中間特徵採用共享存
相關文章
相關標籤/搜索