日々精進

新しく学んだことを書き留めていきます

SageMakerで学習をしている時のTensorBoardを表示する

Estimatorのinitにtensorboard_output_configという引数があるので、これを設定するとtensorboardのログをS3にアップロードしてくれそうだけど、してくれなかった。。 以下で同じ問題で困っている人がいるけど解決してないっぽい。

stackoverflow.com

しょうがないのでtensorboard callbackのlog_dirにS3のパスを指定してアップロードした。詳細は以下参照。

dev.classmethod.jp

SageMaker Studio上でTensorBoardを表示する手順は以下。

docs.aws.amazon.com