fix example typo (#212)

Signed-off-by: Zhang, Weiwei1 <[email protected]>
intel · Aug 7, 2024 · 77a7e60 · 77a7e60
1 parent 1dd2708
commit 77a7e60
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/examples/language-modeling/main.py b/examples/language-modeling/main.py
@@ -44,7 +44,7 @@
     parser.add_argument("--iters", default=200, type=int,
                         help=" iters")
 
-    parser.add_argument("--dataset", default="calib_data/pile-10k", type=str,
+    parser.add_argument("--dataset", default="NeelNanda/pile-10k", type=str,
                         help="The dataset for quantization training. It can be a custom one.")
 
     parser.add_argument("--enable_quanted_input", action='store_true',