DeepSpeed
Generalize the model input format of the flops profiler get_model_profile() API
#1768

Merged

Commits

generalize profiler model input format

cli99 committed 4 years ago
fix format

cli99 committed 4 years ago
Merge branch 'master' into fp-model-input

cli99 committed 4 years ago
fix timining of module forward

cli99 committed 4 years ago
Merge branch 'master' into fp-model-input

tjruwase committed 4 years ago
Merge branch 'master' into fp-model-input

jeffra committed 4 years ago
Merge branch 'master' into fp-model-input

cli99 committed 4 years ago
fix typo

cli99 committed 4 years ago
fix typo

cli99 committed 4 years ago

Loading