You cannot select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
Stanislas0 a3347b25b3 add save_name 2 years ago
..
data Fixed typo: begining -> beginning 2 years ago
model Merge branch 'main' into develop 2 years ago
mpu Fixed typo: thier -> their 2 years ago
optimizer Add megatron pretraining 2 years ago
tokenizer Update model parallelism 2 years ago
tools add finetune codegeex 2 years ago
__init__.py Release cross-platform source code and weights 2 years ago
arguments.py add arguments 2 years ago
checkpointing.py Merge branch 'main' into develop 2 years ago
code_generation_utils.py Refactor beam search 2 years ago
convert_ckpt_parallel.py Fix parallel ckpt conversion 2 years ago
enums.py Add megatron pretraining 2 years ago
global_vars.py Add megatron pretraining 2 years ago
inference.py Release cross-platform source code and weights 2 years ago
initialize.py Release cross-platform source code and weights 2 years ago
learning_rates.py Add megatron pretraining 2 years ago
memory.py Release cross-platform source code and weights 2 years ago
merge_ckpt_parallel.py add save_name 2 years ago
microbatches.py Add megatron pretraining 2 years ago
mindspore_to_megatron.py Release cross-platform source code and weights 2 years ago
p2p_communication.py Release cross-platform source code and weights 2 years ago
schedules.py Add megatron pretraining 2 years ago
training.py Merge branch 'main' into develop 2 years ago
utils.py Add megatron pretraining 2 years ago