mediapipe:モデル学習

差分

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

この比較画面へのリンク

両方とも前のリビジョン 前のリビジョン
次のリビジョン
前のリビジョン
最新のリビジョン両方とも次のリビジョン
mediapipe:モデル学習 [2023/09/30 22:28] – [Multiple GPU] 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( hparams = object_detector.HParams(
   export_dir='exported_model',   export_dir='exported_model',
行 65: 行 65:
   distribution_strategy='mirrored'   distribution_strategy='mirrored'
   )   )
-```+</code>
 ====== 実行 ====== ====== 実行 ======
 うまくいくとモデルが完成するので,出来上がったモデルを試すページは以下 うまくいくとモデルが完成するので,出来上がったモデルを試すページは以下
  • /home/users/2/lolipop.jp-4404d470cd64c603/web/ws/data/pages/mediapipe/モデル学習.txt
  • 最終更新: 2023/10/02 15:14
  • by baba