gcloud ai平台本地火车未在jupyter笔记本中运行

时间:2019-08-29 01:05:18

标签: google-cloud-platform jupyter-notebook gcloud gcp-ai-platform-training

这是另一个帖子中未解决的部分。我正在尝试提交可训练mnist位数的cnn模型的Google云工作。

这是我的系统。 Windows 10,Anaconda,Jupyter Notebook 6,Python 3.6,TF 1.13.0。

我将gcloud命令用于本地火车。第二个单元格似乎停留在[*]状态,直到我关闭并停止ipynb文件之前什么都没有显示。培训紧随其后开始,结果是正确的,因为我在Tensorboard上对其进行了监视。

我可以使其在终端中运行而不会出现此问题。我还成功地将作业提交到了云,并成功完成了。

有没有想到当地的火车问题?代码在这里。

OUTDIR='trained_test'

INPDIR='..\data'
shutil.rmtree(path = OUTDIR, ignore_errors = True) 
!gcloud ai-platform local train \
    --module-name=trainer.task \
    --package-path=trainer \
    -- \
    --output_dir=$OUTDIR \
    --input_dir=$INPDIR \
    --epochs=2 \
    --learning_rate=0.001 \
    --batch_size=100

0 个答案:

没有答案