This website requires JavaScript.
Explore
Help
Register
Sign In
github
/
ColossalAI
Watch
1
Star
0
Fork
0
You've already forked ColossalAI
mirror of
https://github.com/hpcaitech/ColossalAI.git
synced
2025-10-21 23:02:07 +00:00
Code
Issues
Packages
Projects
Releases
Wiki
Activity
Files
70814dc22f1ec2708a3e0ee663ca1b12ee64361f
ColossalAI
/
colossalai
/
engine
/
ophooks
History
ver217
7aef75ca42
[zero] add sharded grad and refactor grad hooks for ShardedModel (
#287
)
2022-03-11 15:50:28 +08:00
..
__init__.py
[zero] add sharded grad and refactor grad hooks for ShardedModel (
#287
)
2022-03-11 15:50:28 +08:00
_base_ophook.py
add pytorch hooks (
#179
)
2022-01-25 22:20:54 +08:00
_memtracer_ophook.py
Feature/zero (
#279
)
2022-03-11 15:50:28 +08:00
_shard_grad_ophook.py
[zero] add sharded grad and refactor grad hooks for ShardedModel (
#287
)
2022-03-11 15:50:28 +08:00
_shard_param_ophook.py
Feature/zero (
#279
)
2022-03-11 15:50:28 +08:00
