-
Notifications
You must be signed in to change notification settings - Fork 2
Open
Description
- 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
Labels
No labels