ARTICLE DETAIL

资讯详情

深耕郑州网站建设与运营推广的一线实战洞察。

Invalid argument: Subshape must have computed start >= end since stride is negative, but is 0 and 2

Invalid argument: Subshape must have computed start >= end since stride is negative, but is 0 and 2 keras-yolov3在win10下训练目标检测算法【win10 gpu 3090 cuda_10.0.130_411.31_win10 cudnn-10.0-windows10-x64-v7.6.5.32】报错如下......E tensorflow/core/grappler/optimizers/meta_optimizer.cc:502] layout failed: Invalid argument: Subshape must have computed start end since stride is negative, but is 0 and 2 (computed from start 0 and end 9223372036854775807 over shape with rank 2 and stride-1) ......E tensorflow/core/grappler/optimizers/meta_optimizer.cc:502] remapper failed: Invalid argument: Subshape must have computed start end since stride is negative, but is 0 and 2 (computed from start 0 and end 9223372036854775807 over shape with rank 2 and stride-1)cuda10下载地址CUDA Toolkit 10.0 Archive | NVIDIA DeveloperSelect Target Platform Click on the green buttons that describe your target platform. Only supported platforms will be shown. Operating System Architecture Distribution Version Installer Type Do you want to cross-compile? Yes No Select Host Platform Click on the green buttons that describe your host platform. Only supported platforms will be shown. Operating Systemhttps://developer.nvidia.com/cuda-10.0-download-archive?target_osWindowstarget_archx86_64target_version10target_typeexelocal建议下载 exe(local)安装包虽文件比较大但一般安装没问题。安装默认目录是C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0cudnn下载地址cuDNN Archive | NVIDIA DeveloperNVIDIA cuDNN is a GPU-accelerated library of primitives for deep neural networks.https://developer.nvidia.cn/rdp/cudnn-archive以下是windows的对应关系解决方法model.py中的140,141行改为下面box_xy (K.sigmoid(feats[..., :2]) grid) / K.cast(grid_shape[..., ::-1], K.dtype(feats)) box_wh K.exp(feats[..., 2:4]) * anchors_tensor / K.cast(input_shape[..., ::-1], K.dtype(feats))
返回列表