mirror of https://github.com/hpcaitech/ColossalAI
569357fea0
* add pytorch hooks fix #175 * remove licenses in src code * add gpu memory tracer * replacing print with logger in ophooks. |
||
---|---|---|
.. | ||
__init__.py | ||
registry.py |