DeepSpeed

7f03282c - [debug utils] see_memory_usage fixes (#890)

[debug utils] see_memory_usage fixes (#890) * see_memory_usage fixes * didn't expect pt-1.2 * fix the order of things * fix the order of things