我的配置环境是:
- Tensorflow-gpu1.4.0
- GTX1080ti
- CUDA8.0&Cudnn6.0
- keras2.1
python3.5
在这样的环境中配置好了之后进行训练,在执行下面模块的时候出现报错
model.train(dataset_train, dataset_val,
learning_rate=config.LEARNING_RATE,
epochs=1,
layers='heads')
# Fine tune all layers
model.train(dataset_train, dataset_val,
learning_rate=config.LEARNING_RATE / 10,
epochs=2,
layers="all")
报错信息
~/anaconda2/envs/tf/lib/python3.5/site-packages/mask_rcnn-2.1-py3

在Tensorflow-gpu1.4.0环境下配置Mask-RCNN时遇到报错:'tf.reduce_mean'函数接收到了意外的'keep_dims'参数。问题根源在于关键字命名与Tensorflow版本不匹配。解决方案是在maskRCNN相关文件中找到报错位置,将'keep_dims'改为'_keep_dims',保存并重新安装MaskRCNN,确保修改生效,然后重新启动训练即可避免此错误。注意,直接删除'keep_dims'会导致使用默认值,无法正常训练。
最低0.47元/天 解锁文章
8万+

被折叠的 条评论
为什么被折叠?



