Add the flops profiler which measures the time, number of estimated flops and parameters of each module in a model #664

cli99 merged 1 commit into

deepspeedai:masterdeepspeedai/DeepSpeed:masterfrom

cli99:flops-profile-staging-v2cli99/DeepSpeed:flops-profile-staging-v2Copy head branch name to clipboard