为什么我无法访问 keras.initializers 类？

Question

我正在尝试使用 keras.initializers 类中的 HeUniform() 方法，但是当我运行我的代码时，我得到以下输出：

AttributeError: module 'tensorflow_core.keras.initializers' has no attribute 'HeUniform'

我运行的代码如下：

init = tf.keras.initializers.HeUniform()

我的进口是：

import gym
import tensorflow as tf
import numpy as np
from tensorflow import keras

from collections import deque
import time
import random

我正在使用从 conda 安装的 tensorflow 版本 2.0.0。我尝试升级版本，但我收到更多错误，说 conda 不存在版本 2.2.0 和 2.4.1。请指教。

Answer 1

看看tensorflow api documentation for version 2.0。没有课程tf.keras.initializers.HeUniform。相反，有一个函数tf.keras.initializers.he_uniform。您可以找到函数on GitHub的源代码。

TensorFlow 建议使用pip 安装。过去，我专门为 tensorflow 创建了一个新的 conda 环境。混合 conda 和 pip 包是不好的做法，这就是为什么我只为特定版本的 tensorflow 创建一个新环境。

conda create -n tf2 python=3.8
conda activate tf2
python -m pip install --no-cache-dir tensorflow==2.4.1

请记住，GPU 版本取决于 CUDA 和 cuDNN 的特定版本。有关要求的更多信息，请参阅https://www.tensorflow.org/install/source#gpu。