<DLRMv2> Add return of construct_model in dlrm_main.py

[JunxiChhen]

Inference DLRMv2 on CPU using dlrm_main.py Issue:

model, optimizer, lr_scheduler = construct_model(args)
TypeError: cannot unpack non-iterable NoneType object

This PR is for this issue fixing.