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
..
hooks	Fixed docstring in colossalai (#171)	2022-01-21 10:44:30 +08:00
__init__.py	Develop/experiments (#59)	2021-12-09 15:08:29 +08:00
_trainer.py	add pytorch hooks (#179)	2022-01-25 22:20:54 +08:00