Commit Graph

2 Commits (dfc3fafe89d62639d3b07a3da53a6bb8487fab3f)

Author SHA1 Message Date
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
zbian 404ecbdcc6 Migrated project 2021-10-28 18:21:23 +02:00