差分
このページの2つのバージョン間の差分を表示します。
両方とも前のリビジョン 前のリビジョン 次のリビジョン | 前のリビジョン最新のリビジョン両方とも次のリビジョン | ||
mediapipe:モデル学習 [2023/09/19 14:52] – [学習] baba | mediapipe:モデル学習 [2023/09/30 22:30] – [Multiple GPU] baba | ||
---|---|---|---|
行 55: | 行 55: | ||
等と設定することで、複数gpuを学習に利用できる。ただし、実際にこのオプションを設定して実行すると distribution_strageryが' | 等と設定することで、複数gpuを学習に利用できる。ただし、実際にこのオプションを設定して実行すると distribution_strageryが' | ||
* num_gpus:4, | * num_gpus:4, | ||
- | * distribution_strategy:' | + | * distribution_strategy=' |
- | でOK. | + | でOK.例えばこんな感じ |
+ | < | ||
+ | hparams = object_detector.HParams( | ||
+ | export_dir=' | ||
+ | epochs=120, | ||
+ | num_gpus=4, | ||
+ | distribution_strategy=' | ||
+ | ) | ||
+ | </ | ||
====== 実行 ====== | ====== 実行 ====== | ||
うまくいくとモデルが完成するので,出来上がったモデルを試すページは以下 | うまくいくとモデルが完成するので,出来上がったモデルを試すページは以下 |