Skip to content

ToDoList #2

@Yagami360

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

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