使用MXNet进行模型训练,一直提示如下:
Running performance tests to find the best convolution algorithm, this can take a while... (set the environment variable MXNET_CUDNN_AUTOTUNE_DEFAULT to 0 to disable)
并且保持在这个状态,等待许久都没有反应。
解决办法是直接跳过这里,set the environment variable MXNET_CUDNN_AUTOTUNE_DEFAULT to 0 to disable。
linux:
export MXNET_CUDNN_AUTOTUNE_DEFAULT=0
windows:
set MXNET_CUDNN_AUTOTUNE_DEFAULT=0
在测试脚本内禁用autotune:
import os
os.environ["MXNET_CUDNN_AUTOTUNE_DEFAULT"] = "0"
tip:在Google Colaboratory环境中,需要采用第三种(脚本内禁用的方式),亲测有效。
导致这个问题是由于每次目标检测是图片的尺寸不一致导致的,可以通过修改输入图片的尺寸一致解决这个问题。
参考资料:
https://blog.youkuaiyun.com/sinat_29957455/article/details/92791103