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/context/process_group_initializer
kurisusnowdeng 0b8161fab8
updated tp layers
2 years ago
..
__init__.py
initializer_1d.py
initializer_2d.py
initializer_2p5d.py
initializer_3d.py updated tp layers 2 years ago
initializer_data.py
initializer_model.py
initializer_pipeline.py
initializer_sequence.py
initializer_tensor.py
process_group_initializer.py