You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
ColossalAI/colossalai
HELSON f6178728a0
[gemini] fix init bugs for modules (#2047)
2 years ago
..
_C
amp
auto_parallel
builder
cli
communication
context
device
engine
fx
gemini [Gemini] free and allocate cuda memory by tensor.storage, add grad hook (#2040) 2 years ago
kernel
logging
nn [gemini] add arguments (#2046) 2 years ago
pipeline
registry
tensor
testing [zero] test gradient accumulation (#1964) 2 years ago
trainer
utils [gemini] fix init bugs for modules (#2047) 2 years ago
zero [zero] test gradient accumulation (#1964) 2 years ago
__init__.py [setup] supported conda-installed torch (#2048) 2 years ago
constants.py
core.py
global_variables.py
initialize.py