harusametime / sagemaker-notebooks

Amazon SageMaker examples with Japanese comments: SageMakerのサンプルです

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

keras_tensorflowのノートブックにタイポ

rtaguchi opened this issue · comments

https://github.com/harusametime/sagemaker-notebooks/blob/master/keras_tensorflow/keras_tensorflow_mnist.ipynb

SageMakerのScript Modeを試す際に、上記のノートブックとソースコードを参考にさせて頂きました。ありがとうございます。
些細な点ですが検証中にタイポに気付いたので共有させて頂きます。

ノートブック中のTensorflowのEstimatorを呼び出す際に、ハイパーパラメータで「batch_size」「n_class」(アンダーバー)という項目を渡していますが、スクリプトファイルの方では「batch-size」「n-class」(ハイフン)というパラメータ名になっているため、ノートブックからパラメータが渡せていないのではないでしょうか。

@rtaguchi ありがとうございます!おっしゃる通りで 5fc22c3 でFixしました。