saymrwulf/pytorch
1
0
Fork 0
mirror of https://github.com/saymrwulf/pytorch.git synced 2026-05-14 20:57:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions
pytorch/docs
History
Ke Wen 07d6ab5aa2 [pipelining] Add pipeline schedules (#125975) 
1. Add pipeline schedules:
- GPipe
- 1F1B
- Interleaved 1F1B
- LoopedBFS

2. Add basic forward and backward tests:
test_schedule.py

Pull Request resolved: https://github.com/pytorch/pytorch/pull/125975
Approved by: https://github.com/wconstab
ghstack dependencies: #125729
2024-05-11 21:17:53 +00:00
..
caffe2
cpp Update Jinja to 3.1.3 (#124976) 2024-04-26 02:57:55 +00:00
source [pipelining] Add pipeline schedules (#125975) 2024-05-11 21:17:53 +00:00
.gitignore
libtorch.rst
make.bat
Makefile
README.md
requirements.txt

README.md

Please see the Writing documentation section of CONTRIBUTING.md for details on both writing and building the docs.