这是另一个帖子中未解决的部分。我正在尝试提交可训练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