Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Running dlrm cpu inference ends up using resnet50 #1214

Open
yuyantingzero opened this issue Apr 18, 2024 · 2 comments
Open

Running dlrm cpu inference ends up using resnet50 #1214

yuyantingzero opened this issue Apr 18, 2024 · 2 comments

Comments

@yuyantingzero
Copy link

Ran with cm run script "app mlperf reference inference _dlrm _cpu" --env.CM_RERUN.

From the attached log file, I saw pulling dlrm cm run script "get dlrm src", but when actually running the benchmark, it becames python3 python/main.py --profile resnet50-onnxruntime --mlperf_conf ../../mlperf.conf --model "/home/perfkit/CM/repos/local/cache/0f978e30da8a423a/resnet50_v1.onnx" ...

Any help is appreciated, thanks!

dlrm.txt

@arjunsuresh
Copy link
Contributor

Can you please share the README link you are following here? Dlrm is replaced by dlrm_v2 in the MLPerf Inference and only dlrm_v2 is supported in CM. For dlrm_v2, the download of the criteo dataset should be done manually.

@yuyantingzero
Copy link
Author

This is the repo/script I am looking at.

Upon checking, did use the wrong model name, changing to _dlrm_99 gets me a little bit further. But now i am running into

Traceback (most recent call last):
  File "/home/perfkit/CM/repos/local/cache/ae4e0392ff6d45b1/inference/recommendation/dlrm_v2/pytorch/python/main.py", line 20, in <module>
    import mlperf_loadgen as lg
ModuleNotFoundError: No module named 'mlperf_loadgen'

For data downloading, is there a page I can reference?
I tired cm run script --tags=get,ml-model,dlrm,raw,terabyte,criteo-terabyte,criteo,recommendation from get-ml-model-dlrm-terabyte which downloads a 90GB file tb00_40M.onnx.tar

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants