github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-04-11 22:54:10 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
115bcc0b416cd050326cbd6e42b1cc00a8053a82
ColossalAI/colossalai/builder/builder.py
Jiarui Fang 569357fea0 add pytorch hooks (#179) 
* add pytorch hooks
fix #175

* remove licenses in src code

* add gpu memory tracer

* replacing print with logger in ophooks.
2022-01-25 22:20:54 +08:00

9.5 KiB
Raw Blame History

View Raw
Reference in New Issue View Git Blame Copy Permalink