TensorFlow 圖變換:FoldBatchNorm

TensorFlow 的計算是按圖(Graph)組織的,構建好的圖有時需要根據需要做一些變換(例如將訓練好的模型部署到生產環境時,去除無用的節點),在保證計算結果不變(或近似不變)的情況下優化計算速度或降低內存佔用。Graph Transform Tool【1】是 TensorFlow 提供的一組可以修改 TensorFlow Graph 的工具,使用方便,易於擴展。   使用 Graph Tra
相關文章
相關標籤/搜索