TensorBoard без обратных вызовов для образа докера Keras в SageMaker - PullRequest
0 голосов
/ 11 декабря 2018

Я пытаюсь добавить функциональность 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() в SageMaker, похоже, не поддерживает модели Keras, скомпилированные с обратными вызовами .

Теперь из этого сообщения о проблеме git было описано, что мне нужно сделать для функциональности TensorBoard:

"Вам нужен код внутри контейнера, чтобысохраняйте контрольные точки на S3, и вам необходимо периодически синхронизировать локальный каталог журналов Tensorboard с вашими контрольными точками S3. "

Итак, чтобы подвести итог, включить TensorBoard в SageMaker с помощью этого пользовательского образа док-станции Keras,похоже, мне нужен способ периодической загрузки файла в корзину S3 во время обучения без использования обратных вызовов.Возможно ли это сделать?Я подумывал о том, чтобы попытаться засунуть этот код в пользовательскую функцию потерь, но я не уверен, что так будет.Любая помощь с благодарностью!

...