diff --git a/ce_tests/dygraph/quant/readme.md b/ce_tests/dygraph/quant/readme.md index 4884d6d006991400af64f37100db9aac23502456..33826a72d45c03af4701d1021f232e4437994254 100644 --- a/ce_tests/dygraph/quant/readme.md +++ b/ce_tests/dygraph/quant/readme.md @@ -2,7 +2,7 @@ 安装需要测试的Paddle版本和PaddleSlim版本。 -准备ImageNet数据集,数据集需要满足paddle hapi的要求。假定解压到`/dataset/ILSVRC2012`文件夹,该文件夹下有`train文件夹、val_hapi文件夹、train_list.txt和val_list.txt文件`。如果数据集有问题,可以私聊。 +准备ImageNet数据集,数据集需要满足paddle hapi的要求。假定解压到`/dataset/ILSVRC2012`文件夹,该文件夹下有`train文件夹、val文件夹、train_list.txt和val_list.txt文件`。 通过`export CUDA_VISIBLE_DEVICES=xx`指定需要使用的GPU ID。 diff --git a/ce_tests/dygraph/quant/run_ptq.sh b/ce_tests/dygraph/quant/run_ptq.sh index aabef03abb595056d38e3b499bf82e3972f6c30a..686ee525cf4a85d10540adf2d363c70d03519b60 100644 --- a/ce_tests/dygraph/quant/run_ptq.sh +++ b/ce_tests/dygraph/quant/run_ptq.sh @@ -1,4 +1,4 @@ -data_path="/dataset/ILSVRC2012" +data_path="/root/datasets/ILSVRC2012" quant_batch_num=10 quant_batch_size=10 diff --git a/ce_tests/dygraph/quant/run_qat.sh b/ce_tests/dygraph/quant/run_qat.sh index a100853e73ca2e7f2912b8504c993a40bcbb07bb..2cc0ddd3dea0cde68ee153c24d373cac2fd6ca8b 100644 --- a/ce_tests/dygraph/quant/run_qat.sh +++ b/ce_tests/dygraph/quant/run_qat.sh @@ -1,5 +1,5 @@ -data_path="/dataset/ILSVRC2012" -val_dir="val_hapi" +data_path="/root/datasets/ILSVRC2012" +val_dir="val" epoch=1 lr=0.0001 batch_size=32 diff --git a/ce_tests/dygraph/quant/src/ptq.py b/ce_tests/dygraph/quant/src/ptq.py index 2695ed0a87a8ac23e18a091730425362df79a58d..bb1138dfc280a270ab971e0399d196adcb219739 100644 --- a/ce_tests/dygraph/quant/src/ptq.py +++ b/ce_tests/dygraph/quant/src/ptq.py @@ -100,7 +100,7 @@ if __name__ == '__main__': ) parser.add_argument( '--val_dir', - default="val_hapi", + default="val", help='the dir that saves val images for paddle.Model') # train diff --git a/ce_tests/dygraph/quant/src/qat.py b/ce_tests/dygraph/quant/src/qat.py index e52b2fec1a16d273eced70bba70b510bfe19e63d..0737505f6a16ddaccba2e6e43fad805386973328 100644 --- a/ce_tests/dygraph/quant/src/qat.py +++ b/ce_tests/dygraph/quant/src/qat.py @@ -154,7 +154,7 @@ if __name__ == '__main__': '(should have subdirectories named "train" and "val"') parser.add_argument( '--val_dir', - default="val_hapi", + default="val", help='the dir that saves val images for paddle.Model') # train diff --git a/ce_tests/dygraph/quant/src/test.py b/ce_tests/dygraph/quant/src/test.py index 1b05b4f00e789e24a7a785c632410f272b45ddae..0bd432ac7d8393308faf699ac96209ef2e7d3581 100644 --- a/ce_tests/dygraph/quant/src/test.py +++ b/ce_tests/dygraph/quant/src/test.py @@ -35,7 +35,7 @@ def eval(args): # prepare data val_dataset = dataset.ImageNetDataset( - path=os.path.join(args.data_dir, 'val_hapi'), mode='val') + path=os.path.join(args.data_dir, 'val'), mode='val') eval_loader = paddle.io.DataLoader( val_dataset, batch_size=args.batch_size, num_workers=5)