To run the DLRM baseline open run_dlrm_baseline.sh

You can change the training parameters.

Run the DLRM training by

   !./run_dlrm_baseline.sh

Observe the overall Foward, Backward and Optimizer time.

Updated: