Yagami360 / MachineLearning_Exercises_Python_PyTorch

機械学習アルゴリズムの PyTorch 実装の練習コード集。

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

ToDoList

Yagami360 opened this issue · comments

  • docker に対応
    • docker-compose に対応
    • 適切な version の python=3.6 をインストールした docker image の作成
    • 適切な version の pytorch=1.1.0 をインストールした docker image の作成
    • docker run(コンテナ立ち上げ) とコンテナ内での python train.py を一括して行えるシェルスクリプトの作成
    • イメージの作成とコンテナ起動を別のマシンで行っても動作するかの検証(dockerfile のユーザー ubuntu(1000) とホスト環境側のユーザーが異なる場合の挙動確認)
  • save_checkpoint(), load_checkpoint() メソッドで step 数を書き込み or 読み込みしないように修正。step 数の書き込み or 読み込みは、save_checkpoint_w_step(), load_checkpoint_w_step() メソッドで行うように修正
  • TensorBoard の表示 step 数が、iteration ではなく学習ステップ数になるように修正
  • 複数 GPU で処理を行えるように修正