├── .gitignore ├── README.md └── appendix ├── PytorchBugs.md ├── module ├── .DS_Store ├── data │ └── data_aug.py ├── models │ ├── classification.md │ ├── fpn.py │ ├── mnasnet.py │ ├── mobilenet_v1.py │ ├── resnet.py │ ├── senet.py │ ├── shufflenet_v2.py │ └── unet.py └── optimizer │ └── Learning_rate.ipynb ├── production ├── distributed │ ├── pytorch-distributed-example-master │ │ ├── .gitignore │ │ ├── LICENSE │ │ ├── README.md │ │ ├── mnist │ │ │ ├── Dockerfile │ │ │ ├── README.md │ │ │ ├── docker-compose-gpu.yml │ │ │ ├── docker-compose.yml │ │ │ └── main.py │ │ ├── setup.cfg │ │ └── toy │ │ │ ├── README.md │ │ │ └── main.py │ └── pytorch-distributed-master │ │ ├── .gitignore │ │ ├── LICENSE │ │ ├── README.md │ │ ├── apex_distributed.py │ │ ├── dataparallel.py │ │ ├── distributed.py │ │ ├── distributed_slurm_main.py │ │ ├── horovod_distributed.py │ │ ├── multiprocessing_distributed.py │ │ ├── requirements.txt │ │ ├── start.sh │ │ └── statistics.sh └── inference │ ├── TensorRT │ ├── .ipynb_checkpoints │ │ └── pytorch_onnx_trt-checkpoint.ipynb │ ├── README.md │ ├── cat.jpeg │ ├── pytorch_onnx_trt.ipynb │ ├── trt_helper.py │ └── trt_int8_calibration_helper.py │ └── flask-api │ ├── .gitignore │ ├── LICENSE │ ├── README.md │ ├── app.py │ ├── cat.jpeg │ ├── imagenet_class_index.json │ └── requirements.txt └── template ├── .gitignore ├── LICENSE ├── README.md ├── assets └── images │ ├── 01.JPG │ └── 02.JPG ├── data ├── test │ ├── .gitkeep │ ├── high │ │ ├── 1.png │ │ └── 22.png │ └── low │ │ ├── 1.png │ │ └── 22.png ├── train │ ├── high │ │ ├── 2.png │ │ └── 5.png │ └── low │ │ ├── 2.png │ │ └── 5.png └── valid │ ├── high │ ├── 1.png │ └── 22.png │ └── low │ ├── 1.png │ └── 22.png ├── datasets.py ├── logs └── .gitignore ├── loss.py ├── metrics.py ├── models.py ├── requirements.txt ├── test.py ├── train.py └── utils.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/polarisZhao/pytorch-cookbook/HEAD/.gitignore -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/polarisZhao/pytorch-cookbook/HEAD/README.md -------------------------------------------------------------------------------- /appendix/PytorchBugs.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/polarisZhao/pytorch-cookbook/HEAD/appendix/PytorchBugs.md -------------------------------------------------------------------------------- /appendix/module/.DS_Store: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/polarisZhao/pytorch-cookbook/HEAD/appendix/module/.DS_Store -------------------------------------------------------------------------------- /appendix/module/data/data_aug.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/polarisZhao/pytorch-cookbook/HEAD/appendix/module/data/data_aug.py -------------------------------------------------------------------------------- /appendix/module/models/classification.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/polarisZhao/pytorch-cookbook/HEAD/appendix/module/models/classification.md -------------------------------------------------------------------------------- /appendix/module/models/fpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/polarisZhao/pytorch-cookbook/HEAD/appendix/module/models/fpn.py -------------------------------------------------------------------------------- /appendix/module/models/mnasnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/polarisZhao/pytorch-cookbook/HEAD/appendix/module/models/mnasnet.py -------------------------------------------------------------------------------- /appendix/module/models/mobilenet_v1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/polarisZhao/pytorch-cookbook/HEAD/appendix/module/models/mobilenet_v1.py -------------------------------------------------------------------------------- /appendix/module/models/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/polarisZhao/pytorch-cookbook/HEAD/appendix/module/models/resnet.py -------------------------------------------------------------------------------- /appendix/module/models/senet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/polarisZhao/pytorch-cookbook/HEAD/appendix/module/models/senet.py -------------------------------------------------------------------------------- /appendix/module/models/shufflenet_v2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/polarisZhao/pytorch-cookbook/HEAD/appendix/module/models/shufflenet_v2.py -------------------------------------------------------------------------------- /appendix/module/models/unet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/polarisZhao/pytorch-cookbook/HEAD/appendix/module/models/unet.py -------------------------------------------------------------------------------- /appendix/module/optimizer/Learning_rate.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/polarisZhao/pytorch-cookbook/HEAD/appendix/module/optimizer/Learning_rate.ipynb -------------------------------------------------------------------------------- /appendix/production/distributed/pytorch-distributed-example-master/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/polarisZhao/pytorch-cookbook/HEAD/appendix/production/distributed/pytorch-distributed-example-master/.gitignore -------------------------------------------------------------------------------- /appendix/production/distributed/pytorch-distributed-example-master/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/polarisZhao/pytorch-cookbook/HEAD/appendix/production/distributed/pytorch-distributed-example-master/LICENSE -------------------------------------------------------------------------------- /appendix/production/distributed/pytorch-distributed-example-master/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/polarisZhao/pytorch-cookbook/HEAD/appendix/production/distributed/pytorch-distributed-example-master/README.md -------------------------------------------------------------------------------- /appendix/production/distributed/pytorch-distributed-example-master/mnist/Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/polarisZhao/pytorch-cookbook/HEAD/appendix/production/distributed/pytorch-distributed-example-master/mnist/Dockerfile -------------------------------------------------------------------------------- /appendix/production/distributed/pytorch-distributed-example-master/mnist/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/polarisZhao/pytorch-cookbook/HEAD/appendix/production/distributed/pytorch-distributed-example-master/mnist/README.md -------------------------------------------------------------------------------- /appendix/production/distributed/pytorch-distributed-example-master/mnist/docker-compose-gpu.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/polarisZhao/pytorch-cookbook/HEAD/appendix/production/distributed/pytorch-distributed-example-master/mnist/docker-compose-gpu.yml -------------------------------------------------------------------------------- /appendix/production/distributed/pytorch-distributed-example-master/mnist/docker-compose.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/polarisZhao/pytorch-cookbook/HEAD/appendix/production/distributed/pytorch-distributed-example-master/mnist/docker-compose.yml -------------------------------------------------------------------------------- /appendix/production/distributed/pytorch-distributed-example-master/mnist/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/polarisZhao/pytorch-cookbook/HEAD/appendix/production/distributed/pytorch-distributed-example-master/mnist/main.py -------------------------------------------------------------------------------- /appendix/production/distributed/pytorch-distributed-example-master/setup.cfg: -------------------------------------------------------------------------------- 1 | [yapf] 2 | based_on_style = google 3 | column_limit = 120 4 | -------------------------------------------------------------------------------- /appendix/production/distributed/pytorch-distributed-example-master/toy/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/polarisZhao/pytorch-cookbook/HEAD/appendix/production/distributed/pytorch-distributed-example-master/toy/README.md -------------------------------------------------------------------------------- /appendix/production/distributed/pytorch-distributed-example-master/toy/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/polarisZhao/pytorch-cookbook/HEAD/appendix/production/distributed/pytorch-distributed-example-master/toy/main.py -------------------------------------------------------------------------------- /appendix/production/distributed/pytorch-distributed-master/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/polarisZhao/pytorch-cookbook/HEAD/appendix/production/distributed/pytorch-distributed-master/.gitignore -------------------------------------------------------------------------------- /appendix/production/distributed/pytorch-distributed-master/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/polarisZhao/pytorch-cookbook/HEAD/appendix/production/distributed/pytorch-distributed-master/LICENSE -------------------------------------------------------------------------------- /appendix/production/distributed/pytorch-distributed-master/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/polarisZhao/pytorch-cookbook/HEAD/appendix/production/distributed/pytorch-distributed-master/README.md -------------------------------------------------------------------------------- /appendix/production/distributed/pytorch-distributed-master/apex_distributed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/polarisZhao/pytorch-cookbook/HEAD/appendix/production/distributed/pytorch-distributed-master/apex_distributed.py -------------------------------------------------------------------------------- /appendix/production/distributed/pytorch-distributed-master/dataparallel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/polarisZhao/pytorch-cookbook/HEAD/appendix/production/distributed/pytorch-distributed-master/dataparallel.py -------------------------------------------------------------------------------- /appendix/production/distributed/pytorch-distributed-master/distributed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/polarisZhao/pytorch-cookbook/HEAD/appendix/production/distributed/pytorch-distributed-master/distributed.py -------------------------------------------------------------------------------- /appendix/production/distributed/pytorch-distributed-master/distributed_slurm_main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/polarisZhao/pytorch-cookbook/HEAD/appendix/production/distributed/pytorch-distributed-master/distributed_slurm_main.py -------------------------------------------------------------------------------- /appendix/production/distributed/pytorch-distributed-master/horovod_distributed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/polarisZhao/pytorch-cookbook/HEAD/appendix/production/distributed/pytorch-distributed-master/horovod_distributed.py -------------------------------------------------------------------------------- /appendix/production/distributed/pytorch-distributed-master/multiprocessing_distributed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/polarisZhao/pytorch-cookbook/HEAD/appendix/production/distributed/pytorch-distributed-master/multiprocessing_distributed.py -------------------------------------------------------------------------------- /appendix/production/distributed/pytorch-distributed-master/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/polarisZhao/pytorch-cookbook/HEAD/appendix/production/distributed/pytorch-distributed-master/requirements.txt -------------------------------------------------------------------------------- /appendix/production/distributed/pytorch-distributed-master/start.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/polarisZhao/pytorch-cookbook/HEAD/appendix/production/distributed/pytorch-distributed-master/start.sh -------------------------------------------------------------------------------- /appendix/production/distributed/pytorch-distributed-master/statistics.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/polarisZhao/pytorch-cookbook/HEAD/appendix/production/distributed/pytorch-distributed-master/statistics.sh -------------------------------------------------------------------------------- /appendix/production/inference/TensorRT/.ipynb_checkpoints/pytorch_onnx_trt-checkpoint.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/polarisZhao/pytorch-cookbook/HEAD/appendix/production/inference/TensorRT/.ipynb_checkpoints/pytorch_onnx_trt-checkpoint.ipynb -------------------------------------------------------------------------------- /appendix/production/inference/TensorRT/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/polarisZhao/pytorch-cookbook/HEAD/appendix/production/inference/TensorRT/README.md -------------------------------------------------------------------------------- /appendix/production/inference/TensorRT/cat.jpeg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/polarisZhao/pytorch-cookbook/HEAD/appendix/production/inference/TensorRT/cat.jpeg -------------------------------------------------------------------------------- /appendix/production/inference/TensorRT/pytorch_onnx_trt.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/polarisZhao/pytorch-cookbook/HEAD/appendix/production/inference/TensorRT/pytorch_onnx_trt.ipynb -------------------------------------------------------------------------------- /appendix/production/inference/TensorRT/trt_helper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/polarisZhao/pytorch-cookbook/HEAD/appendix/production/inference/TensorRT/trt_helper.py -------------------------------------------------------------------------------- /appendix/production/inference/TensorRT/trt_int8_calibration_helper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/polarisZhao/pytorch-cookbook/HEAD/appendix/production/inference/TensorRT/trt_int8_calibration_helper.py -------------------------------------------------------------------------------- /appendix/production/inference/flask-api/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/polarisZhao/pytorch-cookbook/HEAD/appendix/production/inference/flask-api/.gitignore -------------------------------------------------------------------------------- /appendix/production/inference/flask-api/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/polarisZhao/pytorch-cookbook/HEAD/appendix/production/inference/flask-api/LICENSE -------------------------------------------------------------------------------- /appendix/production/inference/flask-api/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/polarisZhao/pytorch-cookbook/HEAD/appendix/production/inference/flask-api/README.md -------------------------------------------------------------------------------- /appendix/production/inference/flask-api/app.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/polarisZhao/pytorch-cookbook/HEAD/appendix/production/inference/flask-api/app.py -------------------------------------------------------------------------------- /appendix/production/inference/flask-api/cat.jpeg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/polarisZhao/pytorch-cookbook/HEAD/appendix/production/inference/flask-api/cat.jpeg -------------------------------------------------------------------------------- /appendix/production/inference/flask-api/imagenet_class_index.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/polarisZhao/pytorch-cookbook/HEAD/appendix/production/inference/flask-api/imagenet_class_index.json -------------------------------------------------------------------------------- /appendix/production/inference/flask-api/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/polarisZhao/pytorch-cookbook/HEAD/appendix/production/inference/flask-api/requirements.txt -------------------------------------------------------------------------------- /appendix/template/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/polarisZhao/pytorch-cookbook/HEAD/appendix/template/.gitignore -------------------------------------------------------------------------------- /appendix/template/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/polarisZhao/pytorch-cookbook/HEAD/appendix/template/LICENSE -------------------------------------------------------------------------------- /appendix/template/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/polarisZhao/pytorch-cookbook/HEAD/appendix/template/README.md -------------------------------------------------------------------------------- /appendix/template/assets/images/01.JPG: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/polarisZhao/pytorch-cookbook/HEAD/appendix/template/assets/images/01.JPG -------------------------------------------------------------------------------- /appendix/template/assets/images/02.JPG: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/polarisZhao/pytorch-cookbook/HEAD/appendix/template/assets/images/02.JPG -------------------------------------------------------------------------------- /appendix/template/data/test/.gitkeep: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /appendix/template/data/test/high/1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/polarisZhao/pytorch-cookbook/HEAD/appendix/template/data/test/high/1.png -------------------------------------------------------------------------------- /appendix/template/data/test/high/22.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/polarisZhao/pytorch-cookbook/HEAD/appendix/template/data/test/high/22.png -------------------------------------------------------------------------------- /appendix/template/data/test/low/1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/polarisZhao/pytorch-cookbook/HEAD/appendix/template/data/test/low/1.png -------------------------------------------------------------------------------- /appendix/template/data/test/low/22.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/polarisZhao/pytorch-cookbook/HEAD/appendix/template/data/test/low/22.png -------------------------------------------------------------------------------- /appendix/template/data/train/high/2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/polarisZhao/pytorch-cookbook/HEAD/appendix/template/data/train/high/2.png -------------------------------------------------------------------------------- /appendix/template/data/train/high/5.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/polarisZhao/pytorch-cookbook/HEAD/appendix/template/data/train/high/5.png -------------------------------------------------------------------------------- /appendix/template/data/train/low/2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/polarisZhao/pytorch-cookbook/HEAD/appendix/template/data/train/low/2.png -------------------------------------------------------------------------------- /appendix/template/data/train/low/5.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/polarisZhao/pytorch-cookbook/HEAD/appendix/template/data/train/low/5.png -------------------------------------------------------------------------------- /appendix/template/data/valid/high/1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/polarisZhao/pytorch-cookbook/HEAD/appendix/template/data/valid/high/1.png -------------------------------------------------------------------------------- /appendix/template/data/valid/high/22.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/polarisZhao/pytorch-cookbook/HEAD/appendix/template/data/valid/high/22.png -------------------------------------------------------------------------------- /appendix/template/data/valid/low/1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/polarisZhao/pytorch-cookbook/HEAD/appendix/template/data/valid/low/1.png -------------------------------------------------------------------------------- /appendix/template/data/valid/low/22.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/polarisZhao/pytorch-cookbook/HEAD/appendix/template/data/valid/low/22.png -------------------------------------------------------------------------------- /appendix/template/datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/polarisZhao/pytorch-cookbook/HEAD/appendix/template/datasets.py -------------------------------------------------------------------------------- /appendix/template/logs/.gitignore: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /appendix/template/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/polarisZhao/pytorch-cookbook/HEAD/appendix/template/loss.py -------------------------------------------------------------------------------- /appendix/template/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/polarisZhao/pytorch-cookbook/HEAD/appendix/template/metrics.py -------------------------------------------------------------------------------- /appendix/template/models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/polarisZhao/pytorch-cookbook/HEAD/appendix/template/models.py -------------------------------------------------------------------------------- /appendix/template/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/polarisZhao/pytorch-cookbook/HEAD/appendix/template/requirements.txt -------------------------------------------------------------------------------- /appendix/template/test.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /appendix/template/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/polarisZhao/pytorch-cookbook/HEAD/appendix/template/train.py -------------------------------------------------------------------------------- /appendix/template/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/polarisZhao/pytorch-cookbook/HEAD/appendix/template/utils.py --------------------------------------------------------------------------------