search

ai-computing / aicomp

Other
13 stars 1 forks source link

1F1B scheduling support #6

Closed ememos closed 4 days ago

ememos commented 7 months ago

In addition to gpipe scheduling, 1F1B pipeline scheduling is also required.

ememos commented 6 months ago

Schedule1F1B class added.

ememos commented 4 days ago

To use the 1f1b schedule, set mode="1f1b" in the run() method of the Optimus_p class.