[debug utils] see_memory_usage fixes#890

tjruwase merged 4 commits intodeepspeedai:masterdeepspeedai/DeepSpeed:masterfrom

stas00:see_memory_usage_improvedstas00/DeepSpeed:see_memory_usage_improvedCopy head branch name to clipboard