PyTorch 実装の練習コード集。
nvidia 製 GPU 搭載マシンでの動作を想定しています。
- Ubuntu :
- シェルスクリプト
.sh
のみ Ubuntu での動作を想定しています。
- シェルスクリプト
- Python : 3.6
- Anaconda :
- PyTorch : 1.x 系
- tensorboardx :
- tqdm :
- imageio :
- Pillow < 7.0.0
nvidia-docker2 で動作します。
- Docker イメージの作成 & Docker コンテナの起動(docker-compose を使用する場合)
$ docker-compose up -d $ docker exec -it -u $(id -u $USER):$(id -g $USER) ml_exercises_pytorch_container /bin/bash
- Deep Neural Network(基礎モデル)
- CNN 系
- GANs
- Graph Convolutional Networks (GCN)
- Transfomer
- 正則化層
- Attention 機構
- データオーギュメント
- 強化学習