TARTRL/PyTorch_Tutorial

PyTorch使用技巧和教程

PythonApache-2.0

PyTorch_Tutorial

0. 环境设置

Ubuntu or Red Hat
Python3.8
pip install -r requirements.txt

1. Pytorch 2.0 初探

这部分主要讲torch.comple()函数的使用。详细介绍见: 知乎:PyTorch 2.0初探。以下是一些测试代码，进入./compile/文件后通过python xxx.py进行运行：

运行测试简单函数: python test_simple.py
运行测试resnet50(mode='default'): python test_resnet50.py
运行测试resnet50(mode='reduce-overhead'): python test_resnet50_optimized.py
运行Huggingface上的BERT模型: python test_bert.py

Citing PyTorch_Tutorial

If you use PyTorch_Tutorial in your work, please cite us:

@article{tartrl2023ptt,
    title={PyTorch Tutorial},
    author={TARTRL Contributors},
    year={2023},
    howpublished={\url{https://github.com/TARTRL/PyTorch_Tutorial}},
}