From c5694896743664cfefbe0c36b2fef90112d5f334 Mon Sep 17 00:00:00 2001 From: l00486551 Date: Wed, 30 Jun 2021 16:54:34 +0800 Subject: [PATCH] fix eval.py data_path conflict --- model_zoo/official/cv/tinydarknet/eval.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/model_zoo/official/cv/tinydarknet/eval.py b/model_zoo/official/cv/tinydarknet/eval.py index c63fdeb81be..68df4ad87fc 100644 --- a/model_zoo/official/cv/tinydarknet/eval.py +++ b/model_zoo/official/cv/tinydarknet/eval.py @@ -98,11 +98,11 @@ def run_eval(): if config.dataset_name == "imagenet": if not cfg.use_label_smooth: cfg.label_smooth_factor = 0.0 - dataset = create_dataset_imagenet(cfg.data_path, 1, False) + dataset = create_dataset_imagenet(cfg.val_data_dir, 1, False) loss = CrossEntropySmooth(sparse=True, reduction="mean", smooth_factor=cfg.label_smooth_factor, num_classes=cfg.num_classes) elif config.dataset_name == "cifar10": - dataset = create_dataset_cifar(dataset_path=config.data_path, + dataset = create_dataset_cifar(dataset_path=config.val_data_dir, do_train=True, repeat_num=1, batch_size=config.batch_size,