文件名称:cuda-convnet:从亚历克斯在谷歌代码上的代码开始
文件大小:1.92MB
文件格式:ZIP
更新时间:2024-07-31 00:01:04
C
NIN 现在在 Caffe Model Zoo cuda-convnet 从亚历克斯在谷歌代码上的代码开始 使用此代码运行 NIN 我在这段代码中实现了 cccp(可级联跨通道参数)池化。 NIN 结构在我在 ICLR2014 上提交的论文:Network In Network 中。 编译代码 要编译此代码,需要 cuda-5.0 或 cuda-5.5。 列出其他依赖项。 在 Kernel 和 PluginSrc 下的 build.sh 脚本中设置路径。 在主目录下运行./build.sh。 将创建一个 dist 目录,其中包含所有 python 代码并在其中构建共享库。 将目录更改为 dist,并将当前路径添加到 LD_LIBRARY_PATH export LD_LIBRARY_PATH= $LD_LIBRARY_PATH :./ 所有的实验都在 dist 目录下运行。