Slide 10
Slide 10 text
TensorflowにおけるTPU実⾏コード
• 基本は TPUEstimator というものを⽤いる。
• model_fn 内でmodelを定義。TRAIN, EVAL, PREDICT時などで処理を変える必要あり。
• Estimatorに⼊⼒する input_fn を tf.data APIで定義。
• (features, label) の出⼒にする必要あり。TFRecordから読み込む事が多い。
• 慣れていないとめんどくさそう。。。(公式のExample (↓)も読みにくい)
10
https://github.com/tensorflow/tpu/blob/master/models/official/resnet/resnet_main.py