Py学习  »  docker

sagemaker中keras-docker图像的无回调tensorboard

Austin • 6 年前 • 699 次点击  

我正试图将tensorboard功能添加到这个sagemaker示例中: https://github.com/awslabs/amazon-sagemaker-examples/blob/master/hyperparameter_tuning/keras_bring_your_own/hpo_bring_your_own_keras_container.ipynb

问题是圣贤的 Estimator.fit() does not seem to support Keras models compiled with callbacks 是的。

从现在开始 this git issue post 有人说,我需要做的是tensorboard功能,

“您需要在容器中输入代码,以便将检查点保存到s3, 您需要定期同步本地tensorboard日志目录 用你的S3检查点。”

总而言之,要在sagemaker中启用tensorboard,使用这个自定义keras docker图像,我需要一种方法,在训练期间定期将文件上传到s3存储桶,而不使用回调。这有可能吗?我正在考虑把这段代码塞进一个自定义的loss函数中,但我不确定这是否是实现它的方法。非常感谢您的帮助!

Python社区是高质量的Python/Django开发社区
本文地址:http://www.python88.com/topic/48273