« tensorflow ３つの方法 | メイン | 写真のソート »
2022年10月15日
tensorflow GPU複数使い
シーケンシャルモデルの作成とコンパイルをMirroredStrategyの中に入れる
mirrored_strategy = tf.distribute.MirroredStrategy()
with mirrored_strategy.scope():
model = tf.keras.Sequential([tf.keras.layers.Dense(1, input_shape=(1,))])
model.compile(loss='mse', optimizer='sgd')
model.fitはMirrored Strategyに入れない
◆Subclassing APの場合Iは？？？
ネット探しても出てこないなぁ。
投稿者 muuming : 2022年10月15日 06:27