ColossalAI/tests/test_config
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
..
sample_config.py Develop/experiments (#59) 2021-12-09 15:08:29 +08:00
test_load_config.py add pytorch hooks (#179) 2022-01-25 22:20:54 +08:00