zoukankan      html  css  js  c++  java
  • Caffe测试单独的算子

    最近有一个需求是测试单独算子在CPU、Caffe使用的GPU、cuDNN上的性能,一个是使用caffe的time问题,还有一个是使用单独的test功能。

    time选项的使用,大家都比较熟悉,单独的test功能,需要专门设置一下。

    上次编译Caffe的博客中提到https://www.cnblogs.com/jourluohua/p/9191322.html

    在make all编译之后,有一个make test是用来编译test功能的,使用test功能的话,需要使用make runtest命令,该命令在Makefile文件中如下:

    runtest: $(TEST_ALL_BIN)
            $(TOOL_BUILD_DIR)/caffe
            $(TEST_ALL_BIN) $(TEST_GPUID) --gtest_shuffle $(TEST_FILTER)

    这里懂Makefile语法规则的人应该不难猜到,是一次运行所有的test功能,但是假如只想测试单独算子的话,测试其他算子没有必要,因此对Makefile进行了修改,比如测试convolution的话,可以写成这样:

    #add by jourluohua
    TEST_CONV = .build_release/test/test_convolution_layer.testbin
    runtest_conv: $(TEST_ALL_BIN)
            $(TOOL_BUILD_DIR)/caffe
            $(TEST_CONV) $(TEST_GPUID) --gtest_shuffle $(TEST_FILTER)

    使用make runtest_conv就可以只测试卷积算子了

  • 相关阅读:
    获取小程序码
    获取目标地与当前地距离
    小程序图片预览
    地图导航
    Jquery无缝滚动
    短信验证
    根据经纬度获取地址
    微信小程序中使用emoji表情
    小程序图片上传
    地球人类的牢笼
  • 原文地址:https://www.cnblogs.com/jourluohua/p/10480195.html
Copyright © 2011-2022 走看看