小编Ada*_*ase的帖子

ValueError:不支持 None 值。代码在 CPU/GPU 上运行正常,但在 TPU 上运行不正常

我正在尝试训练语言翻译模型,并且正在 Google Colab 上的Kaggle Notebookseq2seq中复制粘贴代码。该代码在 CPU 和 GPU 上运行良好,但在 TPU 上训练时出现错误。同样的问题已经在这里被问过。

这是我的代码:

    strategy = tf.distribute.experimental.TPUStrategy(resolver)
    
    with strategy.scope():
      model = create_model()
      model.compile(optimizer = 'rmsprop', loss = 'categorical_crossentropy')
    
    model.fit_generator(generator = generate_batch(X_train, y_train, batch_size = batch_size),
                        steps_per_epoch = train_samples // batch_size,
                        epochs = epochs,
                        validation_data = generate_batch(X_test, y_test, batch_size = batch_size),
                        validation_steps = val_samples // batch_size)
Run Code Online (Sandbox Code Playgroud)

追溯:

Epoch 1/2
---------------------------------------------------------------------------
ValueError                                Traceback (most recent call last)
<ipython-input-60-940fe0ee3c8b> in <module>()
      3                     epochs = epochs,
      4                     validation_data = generate_batch(X_test, y_test, …
Run Code Online (Sandbox Code Playgroud)

python machine-learning deep-learning tensorflow tpu

5
推荐指数
1
解决办法
793
查看次数

为什么创建镜像时docker找不到TensorFlow CPU 2.7.0?

我正在使用以下 Dockerfile 构建 Docker 映像:

FROM python:3.8-alpine
EXPOSE 5000/tcp
WORKDIR /app
COPY . /app
RUN pip install -r requirements.txt
CMD [ "python", "./app.py" ]
Run Code Online (Sandbox Code Playgroud)

这是我使用的命令:

docker build -t my-language-app:1.0 .
Run Code Online (Sandbox Code Playgroud)

它给出了这个错误:

[+] Building 91.5s (9/9) FINISHED
 => [internal] load build definition from Dockerfile                                                               0.0s
 => => transferring dockerfile: 32B                                                                                0.0s
 => [internal] load .dockerignore                                                                                  0.0s
 => => transferring context: 2B                                                                                    0.0s
 => [internal] load metadata for docker.io/library/python:3.8-alpine                                               3.3s
 => [auth] library/python:pull token for registry-1.docker.io                                                      0.0s
 => [internal] load build context …
Run Code Online (Sandbox Code Playgroud)

python docker alpine-linux tensorflow

3
推荐指数
1
解决办法
3051
查看次数