我正在尝试训练语言翻译模型,并且正在 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) 我正在使用以下 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)