文件名称:Dynamic-Network-Surgery-Caffe-Reimplementation:Caffe重新实现动态网络手术
文件大小:8.45MB
文件格式:ZIP
更新时间:2024-06-01 22:05:23
C++
动态网络手术咖啡因的重新实现 动态网络手术的Caffe重新实现(仅不支持GPU / cuDNN)。 官方仓库位于 主要区别: 我们didn't修剪偏见。 我们使对超参数的选择更加清晰直观。 我们重新调整代码的组织。 您可以在训练期间monitor the change of weights sparsity卷积层和完全连接的层monitor the change of weights sparsity 。 我们重写了原始的卷积层和内积层,而不是创建新的类。 在不修改图层类型的情况下,重用现有的.prototxt会变得更加容易。 如何使用 ? 与原始的Caffe框架相同。 $ make all -j8 # USE_NCCL=1 make all -j8 for multi-GPU support $ ./build/tools/caffe train --weights /Mod