Update on "Adding integration test for FSDP Memory Tracking and Estim…

…ation"


Adds an integration test for `FSDPMemTracker` which will help keep `estimation.py` in sync with `train.py`.

`python test_runner.py test_outputs  --test fsdp2_mem_tracker`

cc: gnadathur 

[ghstack-poisoned]

.ci/docker/requirements.txt

@@ -5,3 +5,4 @@ tensorboard
 sentencepiece
 tiktoken
 blobfile
+tabulate