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
..
gradient_handler	Fixed docstring in colossalai (#171)	2022-01-21 10:44:30 +08:00
ophooks	add pytorch hooks (#179)	2022-01-25 22:20:54 +08:00
schedule	fix pipeline forward return tensors (#176)	2022-01-21 15:46:02 +08:00
__init__.py	Develop/experiments (#59)	2021-12-09 15:08:29 +08:00
_base_engine.py	add pytorch hooks (#179)	2022-01-25 22:20:54 +08:00