Skip to content

ToDoList #2

Description

@Yagami360
  • 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 で処理を行えるように修正

Activity

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Projects

    No projects

      Milestone

      No milestone

      Relationships

      None yet

      Development

      No branches or pull requests

      Issue actions