Searched refs:eval_dataset (Results 1 – 9 of 9) sorted by relevance
| /third_party/mindspore/tests/st/model_zoo_tests/DeepFM/src/ |
| D | callback.py | 34 def __init__(self, model, eval_dataset, auc_metric, eval_file_path): argument
37 self.eval_dataset = eval_dataset
44 out = self.model.eval(self.eval_dataset)
|
| D | deepfm.py | 355 def get_callback_list(self, model=None, eval_dataset=None): argument
375 if eval_dataset is None:
377 … "args eval_dataset is {}".format(self.train_config.eval_callback, eval_dataset))
379 eval_callback = EvalCallBack(model, eval_dataset, auc_metric,
|
| /third_party/mindspore/tests/st/model_zoo_tests/wide_and_deep/python_file_for_ci/ |
| D | callbacks.py | 91 def __init__(self, model, eval_dataset, auc_metric, config, print_per_step=1): argument
97 self.eval_dataset = eval_dataset
119 out = self.model.eval(self.eval_dataset, dataset_sink_mode=(not self.sparse))
|
| /third_party/mindspore/tests/st/networks/models/resnet50/ |
| D | test_resnet50_imagenet.py | 168 eval_dataset = create_dataset(dataset_path=eval_path, do_train=False,
218 output = model.eval(eval_dataset)
261 eval_dataset = create_dataset(dataset_path=eval_path, do_train=False,
285 model.init(dataset, eval_dataset)
298 output = model.eval(eval_dataset)
|
| /third_party/mindspore/tests/st/tbe_networks/ |
| D | test_resnet_cifar_1p.py | 148 eval_dataset = create_dataset(1, training=False)
149 res = model.eval(eval_dataset)
|
| D | resnet_cifar.py | 152 eval_dataset = create_dataset(1, training=False) variable
153 res = model.eval(eval_dataset)
|
| /third_party/mindspore/tests/st/mem_reuse/ |
| D | resnet_cifar_memreuse.py | 146 eval_dataset = create_dataset(1, training=False) variable
147 res = model.eval(eval_dataset)
|
| D | resnet_cifar_normal.py | 146 eval_dataset = create_dataset(1, training=False) variable
147 res = model.eval(eval_dataset)
|
| /third_party/mindspore/tests/ut/python/dataset/ |
| D | test_cache_nomap.py | 1450 eval_dataset = ds.TFRecordDataset(DATA_DIR, SCHEMA_DIR)
1451 eval_dataset = eval_dataset.map(input_columns=["image"], operations=decode_op, cache=eval_cache)
1455 eval_iter = eval_dataset.create_dict_iterator(num_epochs=num_epoch)
1616 eval_dataset = ds.TFRecordDataset(DATA_DIR, SCHEMA_DIR)
1618 eval_dataset = eval_dataset.map(input_columns=["image"], operations=decode_op, cache=eval_cache)
1621 eval_iter = eval_dataset.create_dict_iterator(num_epochs=num_epoch)
|