[DLRM v2] How to modify the default training script of DLRM v2 to train the model with limited GPU memory

[DLRM v2] How to modify the default training script of DLRM v2 to train the model with limited GPU memory #1605