mirror of https://github.com/hpcaitech/ColossalAI
cf6d1c9284
* [cli] fixed multi-node job launching * [cli] fixed a bug in version comparison * [cli] support launching with env var * [cli] fixed multi-node job launching * [cli] fixed a bug in version comparison * [cli] support launching with env var * added docstring * [cli] added extra launch arguments * [cli] added default launch rdzv args * [cli] fixed version comparison * [cli] added docstring examples and requierment * polish docstring * polish code * polish code |
||
---|---|---|
.. | ||
amp | ||
builder | ||
cli | ||
communication | ||
context | ||
engine | ||
gemini | ||
kernel | ||
logging | ||
nn | ||
registry | ||
tensor | ||
testing | ||
trainer | ||
utils | ||
zero | ||
__init__.py | ||
constants.py | ||
core.py | ||
global_variables.py | ||
initialize.py |