This website requires JavaScript.
Explore
关于
Help
Register
Sign In
github
/
ColossalAI
mirror of
https://github.com/hpcaitech/ColossalAI
Watch
1
Star
0
Fork
You've already forked ColossalAI
0
Code
Issues
Projects
Releases
Wiki
Activity
05023ecfee
ColossalAI
/
colossalai
/
amp
/
torch_amp
History
Frank Lee
a4e91bc87f
[bug] fixed grad scaler compatibility with torch 1.8 (
#735
)
2022-04-12 16:04:21 +08:00
..
__init__.py
Refactored docstring to google style
2022-03-29 17:17:47 +08:00
_grad_scaler.py
[bug] fixed grad scaler compatibility with torch 1.8 (
#735
)
2022-04-12 16:04:21 +08:00
torch_amp.py
Refactored docstring to google style
2022-03-29 17:17:47 +08:00