未验证 提交 347644b0 编写于 作者: W whs 提交者: GitHub

Fix ce of quant (#874)

上级 2e36a01d
......@@ -34,8 +34,7 @@ def eval(args):
output_handle = predictor.get_output_handle(output_names[0])
# prepare data
val_dataset = dataset.ImageNetDataset(
path=os.path.join(args.data_dir, 'val'), mode='val')
val_dataset = dataset.ImageNetDataset(args.data_dir, mode='val')
eval_loader = paddle.io.DataLoader(
val_dataset, batch_size=args.batch_size, num_workers=5)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册