mediapipe:モデル学習

差分

このページの2つのバージョン間の差分を表示します。

この比較画面へのリンク

両方とも前のリビジョン 前のリビジョン
次のリビジョン
前のリビジョン
最新のリビジョン両方とも次のリビジョン
mediapipe:モデル学習 [2023/09/19 14:52] – [学習] babamediapipe:モデル学習 [2023/09/30 22:30] – [Multiple GPU] baba
行 55: 行 55:
 等と設定することで、複数gpuを学習に利用できる。ただし、実際にこのオプションを設定して実行すると distribution_strageryが'off'ではだめだよ、というエラーになる。では distribution_strategyをどうすればよいのかが、実は述べられていない。結果としては、github上のソースコードから、サンプルとして、num_gpusを1以上に設定しているサンプルファイルでは、 distribution_strategyに 'mirrored'を設定しているものがあり、これに倣って 'mirrored' にしたら学習が進んだ。まとめると、 等と設定することで、複数gpuを学習に利用できる。ただし、実際にこのオプションを設定して実行すると distribution_strageryが'off'ではだめだよ、というエラーになる。では distribution_strategyをどうすればよいのかが、実は述べられていない。結果としては、github上のソースコードから、サンプルとして、num_gpusを1以上に設定しているサンプルファイルでは、 distribution_strategyに 'mirrored'を設定しているものがあり、これに倣って 'mirrored' にしたら学習が進んだ。まとめると、
  * num_gpus:4,  * num_gpus:4,
- * distribution_strategy:'mirrored' + * distribution_strategy='mirrored' 
-でOK.+でOK.例えばこんな感じ
  
 +<code>
 +hparams = object_detector.HParams(
 +  export_dir='exported_model',
 +  epochs=120,
 +  num_gpus=4,
 +  distribution_strategy='mirrored'
 +  )
 +</code>
 ====== 実行 ====== ====== 実行 ======
 うまくいくとモデルが完成するので,出来上がったモデルを試すページは以下 うまくいくとモデルが完成するので,出来上がったモデルを試すページは以下
  • /home/users/2/lolipop.jp-4404d470cd64c603/web/ws/data/pages/mediapipe/モデル学習.txt
  • 最終更新: 2023/10/02 15:14
  • by baba