TensorBoard在SageMaker中没有用于Keras docker映像的回调

时间:2018-12-10 21:07:26

标签: amazon-s3 keras tensorboard amazon-sagemaker

我正试图在此SageMaker示例中添加TensorBoard功能:https://github.com/awslabs/amazon-sagemaker-examples/blob/master/hyperparameter_tuning/keras_bring_your_own/hpo_bring_your_own_keras_container.ipynb

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

现在从this git issue post开始,我描述了TensorBoard功能需要做的是

  

”“您需要在容器内添加代码以将检查点保存到S3,   并且您需要定期同步本地Tensorboard日志目录   和您的S3检查点。”

总而言之,要使用此自定义Keras docker映像在SageMaker中启用TensorBoard,看来我需要一种在训练过程中不使用回调就定期将文件上传到S3存储桶的方法。这可能吗?我当时正在考虑尝试将此代码放入自定义损失函数中,但是我不确定这是否是解决问题的方法。任何帮助将不胜感激!

0 个答案:

没有答案