Keras 多GPU模式在计算val的准确率时报错

最新推荐文章于 2024-05-19 08:28:05 发布

team79

最新推荐文章于 2024-05-19 08:28:05 发布

阅读量3.6k

点赞数 1

CC 4.0 BY-SA版权

分类专栏： ML and DL DeepLearning学习笔记

本文链接：https://blog.youkuaiyun.com/hzh_0000/article/details/80393759

错误信息：

Error: tensorflow/stream_executor/cuda/cuda_dnn.cc:444] could not convert BatchDescriptor {count: 0 feature_map_count: 32 spatial: 149 149 value_min: 0.000000 value_max: 0.000000 layout: BatchDepthYX} to cudnn tensor descriptor: CUDNN_STATUS_BAD_PARAM

github上的解决方法
这个是由于keras源码的问题：
在keras.preprocessing.image.py中我们可以看到：

class Iterator(Sequence):
    """Base class for image data iterators.
    Every `Iterator` must implement the `_get_batches_of_transformed_samples`
    method.
    # Arguments
        n: Integer, total number of samples in the dataset to loop over.
        batch_size: Integer, size of a batch.
        shuffle: Boolean, whether to shuffle the data between epochs.
        seed: Random seeding for data shuffling.
    """

    def __init__(self, n, batch_size, shuffle, seed):
        self.n = n
        self.batch_size = batch_size
        self.seed = seed