github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-01-29 21:49:54 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
73bff1128892dfcab91690790047ffda4ffa8716
ColossalAI/colossalai/utils/profiler/__init__.py
1SAA 73bff11288 Added profiler communication operations 
Fixed bug for learning rate scheduler
2022-03-11 15:50:28 +08:00

2 lines
79 B
Python
Raw Blame History

from .comm_profiler import enable_communication_prof, communication_prof_show
Reference in New Issue View Git Blame Copy Permalink