我的原因是输入数据不同尺寸进行reshape 导致自动进行autotune,
如果输入只有一个尺寸例如224 224 那么没有问题,只会再第一次加载模型的时候出现这句话,然后很快执行完毕
https://discuss.gluon.ai/t/topic/8128
mxnet模型测试设置 MXNET_CUDNN_AUTOTUNE_DEFAULT to 0 to disable,没有用的原因,但是没有解决
最新推荐文章于 2024-04-07 20:42:46 发布
我的原因是输入数据不同尺寸进行reshape 导致自动进行autotune,
如果输入只有一个尺寸例如224 224 那么没有问题,只会再第一次加载模型的时候出现这句话,然后很快执行完毕
https://discuss.gluon.ai/t/topic/8128