代码之家  ›  专栏  ›  技术社区  ›  Austin

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

  •  0
  • Austin  · 技术社区  · 6 年前

    我正试图将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函数中,但我不确定这是否是实现它的方法。非常感谢您的帮助!

    0 回复  |  直到 6 年前