<DLRMv2> Add return of construct_model in dlrm_main.py

intel / ai-reference-models

Intel® AI Reference Models: contains Intel optimizations for running deep learning workloads on Intel® Xeon® Scalable processors and Intel® Data Center GPUs

Apache License 2.0

674 stars 220 forks source link

<DLRMv2> Add return of construct_model in dlrm_main.py #174

Open JunxiChhen opened 6 months ago

JunxiChhen commented 6 months ago

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.