mindspore-Ecosystem
/
mindspore
21
0
Fork 27
Code Issues Pull Requests 6 Projects Releases Wiki Activity

!19167 fix eval.py data_path conflict 

Merge pull request !19167 from Shawny/tdn_cpu
This commit is contained in:
i-robot 2021-07-01 09:04:53 +00:00 committed by Gitee
parent 0c636df5a1 c569489674
commit ad3bca49a7
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
model_zoo/official/cv/tinydarknet/eval.py
View File

@ -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,