欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

解决  报错 TypeError: interleave() got an unexpected keyword argument ‘deterministic‘

程序员文章站 2022-06-13 10:03:32
...

运行相关tensorflow 训练调用 imagenet_input.py 代码,

    num_parallel_calls=self.num_parallel_calls, deterministic=False)
TypeError: interleave() got an unexpected keyword argument 'deterministic'

查看相关代码 提示  ImageNet input pipeline using tf.data.Dataset."""

调用了interleave 函数中有一项参数为deterministic

    dataset = dataset.interleave(
        fetch_dataset, cycle_length=self.num_parallel_calls,
        num_parallel_calls=self.num_parallel_calls, deterministic=False)

于是查看使用的tensorflow 版本 可以看到该版本没有deterministic 这一项,于是去掉,可正常调用

interleave(
    map_func, cycle_length=AUTOTUNE, block_length=1, num_parallel_calls=None
)
相关标签: Tensorflow