Skip to content

Commit

Permalink
fix dlrm when using incbench (#2084)
Browse files Browse the repository at this point in the history
Signed-off-by: Xin He <[email protected]>
  • Loading branch information
xin3he authored Dec 6, 2024
1 parent 619f77b commit 9118449
Showing 1 changed file with 1 addition and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -80,7 +80,7 @@ function run_tuning {
--save-model ${tuned_checkpoint} --test-freq=2048 --print-auc $ARGS \
--load-model=${input_model} --accuracy_only
elif [[ ${mode} == "performance" ]]; then
incbench --num_cores_per_instance 4 -u $MODEL_SCRIPT \
incbench --num_cores_per_instance 4 $MODEL_SCRIPT \
--raw-data-file=${dataset_location}/day --processed-data-file=${dataset_location}/terabyte_processed.npz \
--data-set=terabyte --benchmark \
--memory-map --mlperf-bin-loader --round-targets=True --learning-rate=1.0 \
Expand Down

0 comments on commit 9118449

Please sign in to comment.