Merge pull request #112 from liligwu/fix-dlrm_ubench_train_embeddingb…

…ag_driver fix dlrm/ubench/dlrm_ubench_train_embeddingbag_driver.py
facebookresearch · Feb 16, 2023 · a99ea2c · a99ea2c
2 parents e61e91b + 0518aa9
commit a99ea2c
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/benchmarks/dlrm/ubench/dlrm_ubench_train_embeddingbag_driver.py b/benchmarks/dlrm/ubench/dlrm_ubench_train_embeddingbag_driver.py
@@ -108,7 +108,6 @@ def run_emb(args, run_dataset):
         L,
         E,
         alpha=args.alpha,
-        weights_precision=args.weights_precision,
         weighted=args.weighted,
     )
     if isIntNTableBatched: