我正试图在此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存储桶的方法。这可能吗?我当时正在考虑尝试将此代码放入自定义损失函数中,但是我不确定这是否是解决问题的方法。任何帮助将不胜感激!